diff --git a/examples/reflection/reflection.ipynb b/examples/reflection/reflection.ipynb
index 253dfd075..494bb4aca 100644
--- a/examples/reflection/reflection.ipynb
+++ b/examples/reflection/reflection.ipynb
@@ -44,7 +44,7 @@
    "metadata": {},
    "outputs": [
     {
-     "name": "stdin",
+     "name": "stdout",
      "output_type": "stream",
      "text": [
       "LANGCHAIN_API_KEY ········\n",
@@ -348,7 +348,7 @@
     "builder.add_edge(START, \"generate\")\n",
     "\n",
     "\n",
-    "def should_continue(state: List[BaseMessage]):\n",
+    "def should_continue(state: State):\n",
     "    if len(state[\"messages\"]) > 6:\n",
     "        # End after 3 iterations\n",
     "        return END\n",