diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
index 1f85a8357..82ba6ff96 100644
--- a/libs/langgraph/tests/test_pregel.py
+++ b/libs/langgraph/tests/test_pregel.py
@@ -7259,8 +7259,8 @@ def test_pydantic_state_update_command() -> None:
     assert graph.invoke({"foo": ""}) == {"foo": None}
 
     class State(BaseModel):
-        foo: str | None = None
-        bar: str | None = None
+        foo: Optional[str] = None
+        bar: Optional[str] = None
 
     def node_a(state: State):
         return State(foo="foo")