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 1bbb779..dd724f9 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
@@ -15,6 +15,7 @@ class PostTest extends UnitTestCase {
     $node->expects($this->once())
       ->method('label')
       ->willReturn('Test post');
+    $node->method('bundle')->willReturn('article');
 
     $post = new Post($node);