diff --git a/tests/Fakes/FakeScheduledNodeRepository.php b/tests/Fakes/FakeScheduledNodeRepository.php
index 82bb522..1125b51 100644
--- a/tests/Fakes/FakeScheduledNodeRepository.php
+++ b/tests/Fakes/FakeScheduledNodeRepository.php
@@ -29,6 +29,19 @@ class FakeScheduledNodeRepository implements ScheduledNodeRepository
         return $scheduledNode;
     }
 
+    public function update(ScheduledNode $node): ScheduledNode
+    {
+        $this->existingScheduledNodes[$node->getId()] = $node;
+
+        return new ScheduledNode(
+            id: $node->getId(),
+            date: $node->getDate(),
+            plan: $node->getPlan(),
+            node: $node->getNode(),
+            completed: $node->getCompleted()
+        );
+    }
+
     public function find(int $id): ?ScheduledNode
     {
         return array_find(