diff --git a/docs/docs/concepts/subgraphs.md b/docs/docs/concepts/subgraphs.md
index 092cb9bb0..6a4aefb23 100644
--- a/docs/docs/concepts/subgraphs.md
+++ b/docs/docs/concepts/subgraphs.md
@@ -59,8 +59,9 @@ The main question when adding subgraphs is how the parent graph and subgraph com
         response = model.invoke(state["subgraph_messages"])
         return {"subgraph_messages": response}
 
-    subgraph_builder = StateGraph(State)
-    subgraph_builder.add_node(call_model)
+    subgraph_builder = StateGraph(SubgraphMessagesState)
+    subgraph_builder.add_node("call_model_from_subgraph", call_model)
+    subgraph_builder.add_edge(START, "call_model_from_subgraph")
     ...
     # highlight-next-line
     subgraph = subgraph_builder.compile()