diff --git a/main.py b/main.py
index 735581a..9793ae4 100644
--- a/main.py
+++ b/main.py
@@ -107,7 +107,7 @@ builder.add_conditional_edges(
 builder.add_conditional_edges(
     "rewrite",
     lambda state: "reflect" if state["round"] < state["max_rounds"] else END,
-    {"reflect": "reflect", "END": END}
+    {"reflect": "reflect", END: END}
 )
 
 graph = builder.compile()