diff --git a/src/oss/langgraph/sql-agent.mdx b/src/oss/langgraph/sql-agent.mdx
index 38d1afbb0f..d824187cd6 100644
--- a/src/oss/langgraph/sql-agent.mdx
+++ b/src/oss/langgraph/sql-agent.mdx
@@ -270,10 +270,10 @@ Putting these steps in dedicated nodes lets us (1) force tool-calls when needed,
 ```python
 from typing import Literal
 
-from langchain.agents import ToolNode
 from langchain.messages import AIMessage
 from langchain_core.runnables import RunnableConfig
 from langgraph.graph import END, START, MessagesState, StateGraph
+from langgraph.prebuilt import ToolNode
 
 
 get_schema_tool = next(tool for tool in tools if tool.name == "sql_db_schema")