fix ref

Author: adhami3310

reflex/vars/base.py

@@ -1135,17 +1135,14 @@ def _as_ref(self) -> Var:
         Returns:
             The reference to the var.
         """
-        from .object import ObjectVar
-
-        refs = Var(
-            _js_expr="refs",
+        return Var(
+            _js_expr=f"refs[{Var.create(str(self))}]",
             _var_data=VarData(
                 imports={
                     f"$/{constants.Dirs.STATE_PATH}": [imports.ImportVar(tag="refs")]
                 }
             ),
-        ).to(ObjectVar, Mapping[str, str])
-        return refs[LiteralVar.create(str(self))]
+        ).to(str)
 
     def js_type(self) -> StringVar:
         """Returns the javascript type of the object.

tests/units/test_state.py

@@ -2819,7 +2819,8 @@ class UnionState(BaseState):
         str(UnionState.c3.c2r.c1.foo) == f'{UnionState.c3!s}?.["c2r"]?.["c1"]?.["foo"]'  # pyright: ignore [reportOptionalMemberAccess]
     )
     assert (
-        str(UnionState.c3.c2r.c1r.foo) == f'{UnionState.c3!s}?.["c2r"]?.["c1r"]?.["foo"]'  # pyright: ignore [reportOptionalMemberAccess]
+        str(UnionState.c3.c2r.c1r.foo)
+        == f'{UnionState.c3!s}?.["c2r"]?.["c1r"]?.["foo"]'  # pyright: ignore [reportOptionalMemberAccess]
     )
     assert str(UnionState.c3i.c2) == f'{UnionState.c3i!s}?.["c2"]'
     assert str(UnionState.c3r.c2) == f'{UnionState.c3r!s}?.["c2"]'