diff --git a/src/oss/langchain/short-term-memory.mdx b/src/oss/langchain/short-term-memory.mdx
index 501a2d3300..a09c76265e 100644
--- a/src/oss/langchain/short-term-memory.mdx
+++ b/src/oss/langchain/short-term-memory.mdx
@@ -694,7 +694,7 @@ def update_user_info(
 @tool
 def greet(
     runtime: ToolRuntime[CustomContext, CustomState]
-) -> str:
+) -> str | Command:
     """Use this to greet the user once you found their info."""
     user_name = runtime.state.get("user_name", None)
     if user_name is None: