diff --git a/tests/cases/integration/Entity/entity.compositePK.phpt b/tests/cases/integration/Entity/entity.compositePK.phpt
index b76f009a..b6605cbf 100644
--- a/tests/cases/integration/Entity/entity.compositePK.phpt
+++ b/tests/cases/integration/Entity/entity.compositePK.phpt
@@ -77,6 +77,7 @@ class EntityCompositePKTest extends DataTestCase
 		$this->orm->clear();
 
 		$res = $this->orm->userStatsX->getBy(['date' => new DateTime('2019-01-01')]);
+		Assert::notNull($res);
 		Assert::same(100, $res->value);
 
 		$res->value = 200;
@@ -86,6 +87,7 @@ class EntityCompositePKTest extends DataTestCase
 		$this->orm->clear();
 
 		$res = $this->orm->userStatsX->getBy(['date' => new DateTime('2019-01-01')]);
+		Assert::notNull($res);
 		Assert::same(200, $res->value);
 
 		Environment::$checkAssertions = false;
diff --git a/tests/inc/model/userStatX/UserStatsXMapper.php b/tests/inc/model/userStatX/UserStatsXMapper.php
index f09022e2..41d12526 100644
--- a/tests/inc/model/userStatX/UserStatsXMapper.php
+++ b/tests/inc/model/userStatX/UserStatsXMapper.php
@@ -5,6 +5,9 @@
 use Nextras\Orm\Mapper\Dbal\DbalMapper;
 
 
+/**
+ * @extends DbalMapper<UserStatX>
+ */
 final class UserStatsXMapper extends DbalMapper
 {
 }
diff --git a/tests/inc/model/userStatX/UserStatsXRepository.php b/tests/inc/model/userStatX/UserStatsXRepository.php
index 1e8884dc..87acc7cf 100644
--- a/tests/inc/model/userStatX/UserStatsXRepository.php
+++ b/tests/inc/model/userStatX/UserStatsXRepository.php
@@ -5,6 +5,9 @@
 use Nextras\Orm\Repository\Repository;
 
 
+/**
+ * @extends Repository<UserStatX>
+ */
 final class UserStatsXRepository extends Repository
 {
 	static function getEntityClassNames(): array