1 files changed, 8 insertions, 6 deletions
diff --git a/cli/src/trading_cli/commands/portfolio.py b/cli/src/trading_cli/commands/portfolio.py
index 4f49894..fd3ebd6 100644
--- a/cli/src/trading_cli/commands/portfolio.py
+++ b/cli/src/trading_cli/commands/portfolio.py
@@ -1,6 +1,6 @@
 import asyncio
 import sys
-from datetime import datetime, timedelta, timezone
+from datetime import UTC, datetime, timedelta
 
 import click
 from rich.console import Console
@@ -17,10 +17,11 @@ def portfolio():
 def show():
     """Show the current portfolio holdings and balances."""
     try:
-        from shared.db import Database
+        from sqlalchemy import select
+
         from shared.config import Settings
+        from shared.db import Database
         from shared.sa_models import PositionRow
-        from sqlalchemy import select
     except ImportError as e:
         click.echo(f"Error: Could not import required modules: {e}", err=True)
         sys.exit(1)
@@ -71,10 +72,11 @@ def show():
 def history(days):
     """Show PnL history for the portfolio."""
     try:
-        from shared.db import Database
+        from sqlalchemy import select
+
         from shared.config import Settings
+        from shared.db import Database
         from shared.sa_models import PortfolioSnapshotRow
-        from sqlalchemy import select
     except ImportError as e:
         click.echo(f"Error: Could not import required modules: {e}", err=True)
         sys.exit(1)
@@ -84,7 +86,7 @@ def history(days):
         db = Database(settings.database_url.get_secret_value())
         await db.connect()
         try:
-            since = datetime.now(timezone.utc) - timedelta(days=days)
+            since = datetime.now(UTC) - timedelta(days=days)
             stmt = (
                 select(PortfolioSnapshotRow)
                 .where(PortfolioSnapshotRow.snapshot_at >= since)