diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
index f5dbd838..bab279cb 100644
--- a/vision_agent/agent/vision_agent_coder.py
+++ b/vision_agent/agent/vision_agent_coder.py
@@ -263,7 +263,11 @@ def pick_plan(
             pass
         count += 1
 
-    if best_plan is None:
+    if (
+        best_plan is None
+        or "best_plan" not in best_plan
+        or ("best_plan" in best_plan and best_plan["best_plan"] not in plans)
+    ):
         best_plan = {"best_plan": list(plans.keys())[0]}
 
     if verbosity >= 1:
@@ -820,6 +824,7 @@ def chat_with_workflow(
                 verbosity=self.verbosity,
                 media=media_list,
             )
+            __import__("ipdb").set_trace()
             success = cast(bool, results["success"])
             code = cast(str, results["code"])
             test = cast(str, results["test"])