diff --git a/tests/units/components/test_component.py b/tests/units/components/test_component.py
index e375be414..8d26b02c6 100644
--- a/tests/units/components/test_component.py
+++ b/tests/units/components/test_component.py
@@ -31,7 +31,6 @@ from reflex.utils.exceptions import EventFnArgMismatch
 from reflex.utils.imports import ImportDict, ImportVar, ParsedImportDict, parse_imports
 from reflex.vars import VarData
 from reflex.vars.base import LiteralVar, Var
-from reflex.vars.hooks import HookVar
 
 
 @pytest.fixture
@@ -2078,11 +2077,11 @@ def test_component_add_hooks_var():
                 ),
             ]
 
-    assert list(HookComponent()._get_all_hooks()) == [
-        HookVar.create("const hook3 = useRef(null)"),
+    assert list(hook for hook in HookComponent()._get_all_hooks()) == [
+        "const hook3 = useRef(null)",
         "const hook1 = 42",
         "const hook2 = 43",
-        HookVar.create("useEffect(() => () => {}, [])"),
+        "useEffect(() => () => {}, [])",
     ]
     imports = HookComponent()._get_all_imports()
     assert len(imports) == 1