1 files changed, 11 insertions, 12 deletions

diff --git a/tests/units/Action/TaskAssignCurrentUserTest.php b/tests/units/Action/TaskAssignCurrentUserTest.php
index 2fe84a5d..5569a191 100644
--- a/tests/units/Action/TaskAssignCurrentUserTest.php
+++ b/tests/units/Action/TaskAssignCurrentUserTest.php
@@ -3,10 +3,10 @@
 require_once __DIR__.'/../Base.php';
 
 use Kanboard\Event\GenericEvent;
-use Kanboard\Model\TaskCreation;
-use Kanboard\Model\TaskFinder;
-use Kanboard\Model\Project;
-use Kanboard\Model\Task;
+use Kanboard\Model\TaskCreationModel;
+use Kanboard\Model\TaskFinderModel;
+use Kanboard\Model\ProjectModel;
+use Kanboard\Model\TaskModel;
 use Kanboard\Action\TaskAssignCurrentUser;
 
 class TaskAssignCurrentUserTest extends Base
@@ -15,9 +15,9 @@ class TaskAssignCurrentUserTest extends Base
     {
         $this->container['sessionStorage']->user = array('id' => 1);
 
-        $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 TaskAssignCurrentUserTest extends Base
         $action = new TaskAssignCurrentUser($this->container);
         $action->setProjectId(1);
 
-        $this->assertTrue($action->execute($event, Task::EVENT_CREATE));
+        $this->assertTrue($action->execute($event, TaskModel::EVENT_CREATE));
 
         $task = $taskFinderModel->getById(1);
         $this->assertNotEmpty($task);
@@ -36,9 +36,8 @@ class TaskAssignCurrentUserTest extends Base
 
     public function testWithNoUserSession()
     {
-        $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);
 
         $this->assertEquals(1, $projectModel->create(array('name' => 'test1')));
         $this->assertEquals(1, $taskCreationModel->create(array('project_id' => 1, 'title' => 'test')));
@@ -48,6 +47,6 @@ class TaskAssignCurrentUserTest extends Base
         $action = new TaskAssignCurrentUser($this->container);
         $action->setProjectId(1);
 
-        $this->assertFalse($action->execute($event, Task::EVENT_CREATE));
+        $this->assertFalse($action->execute($event, TaskModel::EVENT_CREATE));
     }
 }