diff --git a/src/index.ts b/src/index.ts
index 15db59cc..44d22ffd 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -79,7 +79,7 @@ const workflow = new StateGraph(GraphAnnotation, GraphConfiguration)
   .addEdge("initialize", "generate-action")
   .addConditionalEdges("generate-action", takeActionOrEnd, ["take-action", END])
   .addEdge("take-action", "progress-plan-step")
-  .addEdge("progress-plan-step", "take-action");
+  .addEdge("progress-plan-step", "generate-action");
 
 // Zod types are messed up
 export const graph = workflow.compile() as any;