diff --git a/tests/Fakes/FakeScheduledNodeRepository.php b/tests/Fakes/FakeScheduledNodeRepository.php
index 1551b3a..f1ce59d 100644
--- a/tests/Fakes/FakeScheduledNodeRepository.php
+++ b/tests/Fakes/FakeScheduledNodeRepository.php
@@ -23,7 +23,17 @@ class FakeScheduledNodeRepository implements ScheduledNodeRepository
         return $scheduledNode;
     }
 
-    public function nextId(): int
+    public function find(int $id): ?ScheduledNode
+    {
+        return array_find(
+            $this->existingScheduledNodes,
+            function (ScheduledNode $scheduledNode) use ($id) {
+                return $scheduledNode->getId() === $id;
+            }
+        );
+    }
+
+    private function nextId(): int
     {
         return count($this->existingScheduledNodes);
     }