diff --git a/tests/units/test_state_tree.py b/tests/units/test_state_tree.py
index aa853af9b..ebdd877de 100644
--- a/tests/units/test_state_tree.py
+++ b/tests/units/test_state_tree.py
@@ -6,12 +6,7 @@ import pytest
 import pytest_asyncio
 
 import reflex as rx
-from reflex.state import (
-    BaseState,
-    StateManager,
-    StateManagerRedis,
-    _substate_key,
-)
+from reflex.state import BaseState, StateManager, StateManagerRedis, _substate_key
 
 
 class Root(BaseState):