diff --git a/docs/docs/agents/multi-agent.md b/docs/docs/agents/multi-agent.md
index d96996a9c..2744442fe 100644
--- a/docs/docs/agents/multi-agent.md
+++ b/docs/docs/agents/multi-agent.md
@@ -384,7 +384,7 @@ Putting this together, here is how you can implement a simple multi-agent system
 ```python
 from typing import Annotated
 from langchain_core.tools import tool, InjectedToolCallId
-from langgraph.prebuilt import create_react_agent, InjectedState
+from langgraph.prebuilt import create_react_agent
 from langgraph.graph import StateGraph, START, MessagesState
 from langgraph.types import Command
 
@@ -636,3 +636,4 @@ Check out LangGraph [supervisor](https://github.com/langchain-ai/langgraph-super
 :::js
 Check out LangGraph [supervisor](https://github.com/langchain-ai/langgraphjs/tree/main/libs/langgraph-supervisor#customizing-handoff-tools) and [swarm](https://github.com/langchain-ai/langgraphjs/tree/main/libs/langgraph-swarm#customizing-handoff-tools) documentation to learn how to customize handoffs.
 :::
+