diff --git a/tests/units/test_state.py b/tests/units/test_state.py
index e223b4f19..dbe92ded4 100644
--- a/tests/units/test_state.py
+++ b/tests/units/test_state.py
@@ -3358,8 +3358,8 @@ def test_fallback_pickle():
     """Test that state serialization will fall back to dill."""
 
     class DillState(BaseState):
-        _o: Obj | None = None
-        _f: Callable | None = None
+        _o: Optional[Obj] = None
+        _f: Optional[Callable] = None
         _g: Any = None
 
     state = DillState(_reflex_internal_init=True)  # type: ignore