diff --git a/tests/data/validators/models/ValidatorTestRefModel.php b/tests/data/validators/models/ValidatorTestRefModel.php
index bf0ebe5dbd..bd70734350 100644
--- a/tests/data/validators/models/ValidatorTestRefModel.php
+++ b/tests/data/validators/models/ValidatorTestRefModel.php
@@ -4,6 +4,11 @@ namespace yiiunit\data\validators\models;
 
 use yiiunit\data\ar\ActiveRecord;
 
+/**
+ * @property int id
+ * @property string a_field
+ * @property int ref
+ */
 class ValidatorTestRefModel extends ActiveRecord
 {
 
diff --git a/tests/framework/validators/UniqueValidatorTest.php b/tests/framework/validators/UniqueValidatorTest.php
index 026cbadda3..1a8708b0fd 100644
--- a/tests/framework/validators/UniqueValidatorTest.php
+++ b/tests/framework/validators/UniqueValidatorTest.php
@@ -101,7 +101,7 @@ class UniqueValidatorTest extends DatabaseTestCase
             'targetAttribute' => ['order_id', 'item_id'],
         ]);
         // validate old record
-        /** @var ValidatorTestMainModel $m */
+        /** @var OrderItem $m */
         $m = OrderItem::findOne(['order_id' => 1, 'item_id' => 2]);
         $val->validateAttribute($m, 'order_id');
         $this->assertFalse($m->hasErrors('order_id'));