diff --git a/src/screener/t_atr_ema.py b/src/screener/t_atr_ema.py
index e8531da..05e4dd3 100644
--- a/src/screener/t_atr_ema.py
+++ b/src/screener/t_atr_ema.py
@@ -1,3 +1,4 @@
+import pandas as pd
 from utils.data_utils import (
     get_stock_data, validate_signal_date, print_signal, 
     save_signals_to_csv, get_qualified_stocks,
diff --git a/src/screener/t_sunnyband.py b/src/screener/t_sunnyband.py
index 990237a..93745dd 100644
--- a/src/screener/t_sunnyband.py
+++ b/src/screener/t_sunnyband.py
@@ -1,3 +1,6 @@
+import pandas as pd
+from datetime import datetime, timedelta
+from db.db_connection import create_client
 from utils.data_utils import (
     get_stock_data, validate_signal_date, print_signal, 
     save_signals_to_csv, get_qualified_stocks,