1 files changed, 11 insertions, 11 deletions

diff --git a/tests/units/Action/TaskAssignColorLinkTest.php b/tests/units/Action/TaskAssignColorLinkTest.php
index 6df029d3..07d0969b 100644
--- a/tests/units/Action/TaskAssignColorLinkTest.php
+++ b/tests/units/Action/TaskAssignColorLinkTest.php
@@ -3,19 +3,19 @@
 require_once __DIR__.'/../Base.php';
 
 use Kanboard\Event\GenericEvent;
-use Kanboard\Model\TaskCreation;
-use Kanboard\Model\TaskFinder;
-use Kanboard\Model\Project;
-use Kanboard\Model\TaskLink;
+use Kanboard\Model\TaskCreationModel;
+use Kanboard\Model\TaskFinderModel;
+use Kanboard\Model\ProjectModel;
+use Kanboard\Model\TaskLinkModel;
 use Kanboard\Action\TaskAssignColorLink;
 
 class TaskAssignColorLinkTest extends Base
 {
     public function testChangeColor()
     {
-        $projectModel = new Project($this->container);
-        $taskCreationModel = new TaskCreation($this->container);
-        $taskFinderModel = new TaskFinder($this->container);
+        $projectModel = new ProjectModel($this->container);
+        $taskCreationModel = new TaskCreationModel($this->container);
+        $taskFinderModel = new TaskFinderModel($this->container);
 
         $this->assertEquals(1, $projectModel->create(array('name' => 'test1')));
         $this->assertEquals(1, $taskCreationModel->create(array('project_id' => 1, 'title' => 'test')));
@@ -27,7 +27,7 @@ class TaskAssignColorLinkTest extends Base
         $action->setParam('color_id', 'red');
         $action->setParam('link_id', 1);
 
-        $this->assertTrue($action->execute($event, TaskLink::EVENT_CREATE_UPDATE));
+        $this->assertTrue($action->execute($event, TaskLinkModel::EVENT_CREATE_UPDATE));
 
         $task = $taskFinderModel->getById(1);
         $this->assertNotEmpty($task);
@@ -36,8 +36,8 @@ class TaskAssignColorLinkTest extends Base
 
     public function testWithWrongLink()
     {
-        $projectModel = new Project($this->container);
-        $taskCreationModel = new TaskCreation($this->container);
+        $projectModel = new ProjectModel($this->container);
+        $taskCreationModel = new TaskCreationModel($this->container);
 
         $this->assertEquals(1, $projectModel->create(array('name' => 'test1')));
         $this->assertEquals(1, $taskCreationModel->create(array('project_id' => 1, 'title' => 'test')));
@@ -49,6 +49,6 @@ class TaskAssignColorLinkTest extends Base
         $action->setParam('color_id', 'red');
         $action->setParam('link_id', 1);
 
-        $this->assertFalse($action->execute($event, TaskLink::EVENT_CREATE_UPDATE));
+        $this->assertFalse($action->execute($event, TaskLinkModel::EVENT_CREATE_UPDATE));
     }
 }