diff --git a/src/oss/langchain/short-term-memory.mdx b/src/oss/langchain/short-term-memory.mdx
index 80c990e754..e3694cf725 100644
--- a/src/oss/langchain/short-term-memory.mdx
+++ b/src/oss/langchain/short-term-memory.mdx
@@ -981,8 +981,9 @@ def trim_messages(state: AgentState, runtime: Runtime) -> dict[str, Any] | None:
 
 agent = create_agent(
     model,
-    tools=tools,
+    tools=[],
     middleware=[trim_messages]
+    checkpoint=InMemorySaver()
 )
 
 config: RunnableConfig = {"configurable": {"thread_id": "1"}}