diff --git a/src/oss/langchain/agents.mdx b/src/oss/langchain/agents.mdx
index 78590af07c..f43f6d1b1b 100644
--- a/src/oss/langchain/agents.mdx
+++ b/src/oss/langchain/agents.mdx
@@ -690,6 +690,7 @@ Use middleware to define custom state when your custom state needs to be accesse
 ```python
 from langchain.agents import AgentState
 from langchain.agents.middleware import AgentMiddleware
+from typing import Any
 
 
 class CustomState(AgentState):