diff --git a/web/modules/custom/my_module/tests/src/Unit/Entity/PostTest.php b/web/modules/custom/my_module/tests/src/Unit/Entity/PostTest.php
index 841dd29..4ab5273 100644
--- a/web/modules/custom/my_module/tests/src/Unit/Entity/PostTest.php
+++ b/web/modules/custom/my_module/tests/src/Unit/Entity/PostTest.php
@@ -3,12 +3,15 @@
 namespace Drupal\Tests\my_module\Unit\Entity;
 
 use Drupal\my_module\Entity\Post;
+use Drupal\node\NodeInterface;
 use Drupal\Tests\UnitTestCase;
 
 class PostTest extends UnitTestCase {
 
   /** @test */
   public function it_returns_the_title() {
+    $node = $this->createMock(NodeInterface::class);
+
     $post = new Post($node);
 
     $this->assertSame('Test post', $post->getTitle());