diff --git a/docs/docs/tutorials/multi_agent/agent_supervisor.ipynb b/docs/docs/tutorials/multi_agent/agent_supervisor.ipynb
index 9ad48024d..111d0b05e 100644
--- a/docs/docs/tutorials/multi_agent/agent_supervisor.ipynb
+++ b/docs/docs/tutorials/multi_agent/agent_supervisor.ipynb
@@ -135,7 +135,7 @@
     "from typing_extensions import TypedDict\n",
     "\n",
     "from langchain_anthropic import ChatAnthropic\n",
-    "from langgraph.graph import MessagesState\n",
+    "from langgraph.graph import MessagesState, END\n",
     "from langgraph.types import Command\n",
     "\n",
     "\n",