diff --git a/examples/plan-and-execute/plan-and-execute.ipynb b/examples/plan-and-execute/plan-and-execute.ipynb
index c681480ba..0c24e1e2f 100644
--- a/examples/plan-and-execute/plan-and-execute.ipynb
+++ b/examples/plan-and-execute/plan-and-execute.ipynb
@@ -398,7 +398,7 @@
     "\n",
     "\n",
     "def should_end(state: PlanExecute):\n",
-    "    if state[\"response\"]:\n",
+    "    if \"response\" in state and state[\"response\"]:\n",
     "        return True\n",
     "    else:\n",
     "        return False"