diff --git a/tests/__snapshots__/test_pregel.ambr b/tests/__snapshots__/test_pregel.ambr
index 1b1866624..13bf7a8b7 100644
--- a/tests/__snapshots__/test_pregel.ambr
+++ b/tests/__snapshots__/test_pregel.ambr
@@ -1434,6 +1434,9 @@
   '''
   graph TD;
   	__start__ --> Researcher;
+  	subgraph Researcher
+  	Researcher -. redo .-> Researcher;
+  	end
   	Researcher -. continue .-> Chart_Generator;
   	Researcher -. call_tool .-> Call_Tool;
   	Researcher -. end .-> __end__;
diff --git a/tests/test_pregel.py b/tests/test_pregel.py
index c8f20eda6..aedc89a80 100644
--- a/tests/test_pregel.py
+++ b/tests/test_pregel.py
@@ -4653,7 +4653,12 @@ def test_repeat_condition(snapshot: SnapshotAssertion) -> None:
     workflow.add_conditional_edges(
         "Researcher",
         router,
-        {"continue": "Chart Generator", "call_tool": "Call Tool", "end": END},
+        {
+            "redo": "Researcher",
+            "continue": "Chart Generator",
+            "call_tool": "Call Tool",
+            "end": END,
+        },
     )
     workflow.add_conditional_edges(
         "Chart Generator",