1 files changed, 3 insertions, 0 deletions
diff --git a/tests/units/Filter/TaskCommentFilterTest.php b/tests/units/Filter/TaskCommentFilterTest.php
index 95ffb32f..e6e20cbe 100644
--- a/tests/units/Filter/TaskCommentFilterTest.php
+++ b/tests/units/Filter/TaskCommentFilterTest.php
@@ -21,8 +21,10 @@ class TaskCommentFilterTest extends Base
         $this->assertEquals(1, $projectModel->create(array('name' => 'Test')));
         $this->assertEquals(1, $taskCreation->create(array('title' => 'Test', 'project_id' => 1)));
         $this->assertEquals(1, $commentModel->create(array('task_id' => 1, 'user_id' => 1, 'comment' => 'This is a test')));
+        $this->assertEquals(2, $commentModel->create(array('task_id' => 1, 'user_id' => 1, 'comment' => 'This is another test')));
 
         $filter = new TaskCommentFilter();
+        $filter->setDatabase($this->container['db']);
         $filter->withQuery($query);
         $filter->withValue('test');
         $filter->apply();
@@ -43,6 +45,7 @@ class TaskCommentFilterTest extends Base
         $this->assertEquals(1, $commentModel->create(array('task_id' => 1, 'user_id' => 1, 'comment' => 'This is a test')));
 
         $filter = new TaskCommentFilter();
+        $filter->setDatabase($this->container['db']);
         $filter->withQuery($query);
         $filter->withValue('foobar');
         $filter->apply();