diff --git a/test_reserved_keywords_full.py b/test_reserved_keywords_full.py
index 353026684..f7326be1a 100644
--- a/test_reserved_keywords_full.py
+++ b/test_reserved_keywords_full.py
@@ -4,7 +4,7 @@ from langchain_core.messages import AIMessage, AnyMessage
 from langgraph.prebuilt import ToolNode, InjectedState, InjectedStore
 from langgraph.store.memory import InMemoryStore
 from langgraph.store.base import BaseStore
-from langgraph.types import MessagesState
+from langgraph.graph import MessagesState
 
 
 def test_tool_node_inject_runtime_reserved_keyword() -> None:
@@ -179,3 +179,4 @@ if __name__ == "__main__":
     
     print("\nAll tests passed!")
 
+