diff --git a/tests/Fakes/FakeNodeRepository.php b/tests/Fakes/FakeNodeRepository.php
index 0a679f3..71335ff 100644
--- a/tests/Fakes/FakeNodeRepository.php
+++ b/tests/Fakes/FakeNodeRepository.php
@@ -54,7 +54,7 @@ class FakeNodeRepository implements NodeRepository
         return array_filter(
             $this->existingNodes,
             function (Node $node) use ($id) {
-                return $node->getId() === $id;
+                return $node->getText()->getId() === $id;
             }
         );
     }