1 files changed, 6 insertions, 5 deletions
diff --git a/tests/integration/test_strategy_signal_flow.py b/tests/integration/test_strategy_signal_flow.py
index 6b048fb..3f7ec35 100644
--- a/tests/integration/test_strategy_signal_flow.py
+++ b/tests/integration/test_strategy_signal_flow.py
@@ -8,15 +8,16 @@ sys.path.insert(
 )
 sys.path.insert(0, str(Path(__file__).resolve().parents[2] / "services" / "strategy-engine"))
 
-import pytest
+from datetime import UTC, datetime
 from decimal import Decimal
-from datetime import datetime, timezone
 from unittest.mock import AsyncMock
 
-from shared.models import Candle
-from shared.events import CandleEvent
+import pytest
 from strategy_engine.engine import StrategyEngine
 
+from shared.events import CandleEvent
+from shared.models import Candle
+
 
 @pytest.fixture
 def candles():
@@ -28,7 +29,7 @@ def candles():
             Candle(
                 symbol="AAPL",
                 timeframe="1m",
-                open_time=datetime(2025, 1, 1, i, 0, tzinfo=timezone.utc),
+                open_time=datetime(2025, 1, 1, i, 0, tzinfo=UTC),
                 open=price,
                 high=price + 1,
                 low=price - 1,