diff --git a/src/utils/data_utils.py b/src/utils/data_utils.py
index 645d8dc..efa505a 100644
--- a/src/utils/data_utils.py
+++ b/src/utils/data_utils.py
@@ -4,6 +4,7 @@ from datetime import datetime, timedelta
 from trading.position_calculator import PositionCalculator
 from utils.common_utils import get_user_input, get_stock_data, get_qualified_stocks
 from typing import Optional
+from db.db_connection import create_client
 
 def get_float_input(prompt: str) -> Optional[float]:
     return get_user_input(prompt, float)