diff --git a/main.py b/main.py
index 40d2f2e..672df5f 100644
--- a/main.py
+++ b/main.py
@@ -67,7 +67,12 @@ builder.set_entry_point("draft_answer")
 builder.add_edge("draft_answer", "reflect")
 builder.add_conditional_edges(
     "reflect",
-    lambda state: "END" if state["verdict"] == "ok" else "rewrite",
+    lambda state: (
+        "END"
+        if state["verdict"] == "ok"
+        or state.get("round", 0) >= state.get("max_rounds", 2)
+        else "rewrite"
+    ),
 )
 builder.add_edge("rewrite", "reflect")