diff --git a/web/modules/custom/my_module/src/Entity/Post.php b/web/modules/custom/my_module/src/Entity/Post.php
new file mode 100644
index 0000000..9690518
--- /dev/null
+++ b/web/modules/custom/my_module/src/Entity/Post.php
@@ -0,0 +1,5 @@
+<?php
+
+namespace Drupal\my_module\Entity;
+
+class Post {}
diff --git a/web/modules/custom/my_module/tests/src/Kernel/ArticleRepositoryTest.php b/web/modules/custom/my_module/tests/src/Kernel/ArticleRepositoryTest.php
index f3af15e..443486b 100644
--- a/web/modules/custom/my_module/tests/src/Kernel/ArticleRepositoryTest.php
+++ b/web/modules/custom/my_module/tests/src/Kernel/ArticleRepositoryTest.php
@@ -4,6 +4,7 @@ namespace Drupal\Tests\my_module\Kernel;
 
 use Drupal\Core\Datetime\DrupalDateTime;
 use Drupal\KernelTests\Core\Entity\EntityKernelTestBase;
+use Drupal\my_module\Entity\Post;
 use Drupal\my_module\Repository\ArticleRepository;
 use Drupal\node\NodeInterface;
 use Drupal\Tests\node\Traits\NodeCreationTrait;
@@ -23,9 +24,8 @@ class ArticleRepositoryTest extends EntityKernelTestBase {
 
     $this->assertCount(1, $articles);
     $this->assertIsObject($articles[1]);
-    $this->assertInstanceOf(NodeInterface::class, $articles[1]);
-    $this->assertSame('article', $articles[1]->bundle());
-    $this->assertSame('Test post', $articles[1]->label());
+    $this->assertInstanceOf(Post::class, $articles[1]);
+    $this->assertSame('Test post', $articles[1]->getTitle());
   }
 
   /** @test */