diff --git a/src/trading/trading_plan.py b/src/trading/trading_plan.py
index 931dc8b..33a5b35 100644
--- a/src/trading/trading_plan.py
+++ b/src/trading/trading_plan.py
@@ -137,7 +137,8 @@ def save_trading_plan(plan: TradingPlan) -> int:
         if not plan.id:
             # Generate new ID for new plans
             result = client.query("SELECT max(id) FROM trading_plans")
-            plan.id = (result[0][0] or 0) + 1
+            max_id = result.result_rows[0][0] if result.result_rows else 0
+            plan.id = (max_id or 0) + 1
             plan.created_at = datetime.now()
         
         plan.updated_at = datetime.now()