diff --git a/tests/CommentAddTest.php b/tests/CommentAddTest.php
index 722acb50..670c0b7a 100644
--- a/tests/CommentAddTest.php
+++ b/tests/CommentAddTest.php
@@ -20,7 +20,7 @@ class CommentAddTest extends AbstractTest
 		$this->assert->isNotNull($comment->commentDate);
 		$this->assert->doesNotThrow(function() use ($comment)
 		{
-			UserModel::findById($comment->getId());
+			CommentModel::findById($comment->getId());
 		});
 	}
 
diff --git a/tests/CommentEditTest.php b/tests/CommentEditTest.php
index 1097393e..73a71f32 100644
--- a/tests/CommentEditTest.php
+++ b/tests/CommentEditTest.php
@@ -17,7 +17,7 @@ class CommentEditTest extends AbstractTest
 		$this->assert->isNotNull($comment->commentDate);
 		$this->assert->doesNotThrow(function() use ($comment)
 		{
-			UserModel::findById($comment->getId());
+			CommentModel::findById($comment->getId());
 		});
 	}