diff --git a/tests/units/test_state.py b/tests/units/test_state.py
index 1e17fc653..84a97efb6 100644
--- a/tests/units/test_state.py
+++ b/tests/units/test_state.py
@@ -31,6 +31,7 @@ from pydantic.v1 import BaseModel as BaseModelV1
 
 import reflex as rx
 import reflex.config
+import reflex.utils.console
 from reflex import constants
 from reflex.app import App
 from reflex.base import Base
@@ -1857,7 +1858,7 @@ async def test_state_manager_lock_warning_threshold_contend(
         substate_token_redis: A token + substate name for looking up in state manager.
         mocker: Pytest mocker object.
     """
-    console_warn = mocker.patch("reflex.utils.console.warn")
+    console_warn = mocker.spy(reflex.utils.console, "warn")
 
     state_manager_redis.lock_expiration = LOCK_EXPIRATION
     state_manager_redis.lock_warning_threshold = LOCK_WARNING_THRESHOLD
@@ -1875,7 +1876,7 @@ async def test_state_manager_lock_warning_threshold_contend(
 
     await tasks[0]
     console_warn.assert_called()
-    assert console_warn.call_count == 7
+    assert console_warn.call_count == 1
 
 
 class CopyingAsyncMock(AsyncMock):