diff --git a/tests/units/utils/test_utils.py b/tests/units/utils/test_utils.py
index 7c9449cda..435089e22 100644
--- a/tests/units/utils/test_utils.py
+++ b/tests/units/utils/test_utils.py
@@ -600,7 +600,8 @@ def test_style_prop_with_event_handler_value(callable):
 def cleanup_reflex_env_mode():
     """Cleanup the reflex env mode.
 
-    Yields: None
+    Yields:
+        None
     """
     yield
     EnvironmentVariables.REFLEX_ENV_MODE.set(None)