diff --git a/src/streamlit_app.py b/src/streamlit_app.py
index 759cb5f..e135d00 100644
--- a/src/streamlit_app.py
+++ b/src/streamlit_app.py
@@ -244,7 +244,7 @@ def trading_journal_page():
                     
                     add_trade(trade)
                     st.success("Trade added successfully!")
-                    st.experimental_set_query_params(rerun=True)
+                    st.query_params(rerun=True)
                 except Exception as e:
                     st.error(f"Error adding trade: {str(e)}")
     
@@ -310,7 +310,7 @@ def trading_journal_page():
                     
                     update_trade(trade_id, updates)
                     st.success("Trade updated successfully!")
-                    st.experimental_set_query_params(rerun=True)
+                    st.query_params(rerun=True)
                 except Exception as e:
                     st.error(f"Error updating trade: {str(e)}")
         else:
diff --git a/src/trading/trading_plan.py b/src/trading/trading_plan.py
index 33a5b35..a2f4a8a 100644
--- a/src/trading/trading_plan.py
+++ b/src/trading/trading_plan.py
@@ -267,7 +267,8 @@ def get_all_trading_plans(status: Optional[PlanStatus] = None) -> List[TradingPl
             query += " ORDER BY updated_at DESC"
             
             results = client.query(query, params)
-            return [get_trading_plan(result[0]) for result in results]
+            rows = results.result_rows
+            return [get_trading_plan(row[0]) for row in rows]
         
         except Exception as e:
             print(f"Error retrieving trading plans: {e}")