diff --git a/src/oss/langchain/test.mdx b/src/oss/langchain/test.mdx
index 494cb093ca..2cf4ec5cc0 100644
--- a/src/oss/langchain/test.mdx
+++ b/src/oss/langchain/test.mdx
@@ -58,10 +58,16 @@ agent = create_agent(
 )
 
 # First invocation
-agent.invoke(HumanMessage(content="I live in Sydney, Australia."))
+agent.invoke(
+    {"messages": [HumanMessage(content="I live in Sydney, Australia")]},
+    config={"configurable": {"thread_id": "session-1"}}
+)
 
 # Second invocation: the first message is persisted (Sydney location), so the model returns GMT+10 time
-agent.invoke(HumanMessage(content="What's my local time?"))
+agent.invoke(
+    {"messages": [HumanMessage(content="What's my local time?")]},
+    config={"configurable": {"thread_id": "session-1"}}
+)
 ```
 :::