diff --git a/src/pages/trading/trading_system_page.py b/src/pages/trading/trading_system_page.py
index 18d5ea6..da4c58e 100644
--- a/src/pages/trading/trading_system_page.py
+++ b/src/pages/trading/trading_system_page.py
@@ -9,7 +9,8 @@ import time
 from utils.common_utils import get_stock_data
 from trading.watchlist import (
     create_watchlist, get_watchlists, add_to_watchlist, 
-    remove_from_watchlist, get_watchlist_items, WatchlistItem
+    remove_from_watchlist, get_watchlist_items, WatchlistItem,
+    ensure_tables_exist
 )
 from db.db_connection import create_client