diff --git a/scripts/capital_flow_collector.py b/scripts/capital_flow_collector.py
index adac6e51..22477485 100644
--- a/scripts/capital_flow_collector.py
+++ b/scripts/capital_flow_collector.py
@@ -11,7 +11,7 @@ from datetime import datetime
 from urllib.request import urlopen, Request
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from threading import Semaphore
-from mofin_db import get_conn, write_capital_flow_cache, query_holdings, query_holdings_db, query_watchlist
+from mofin_db import get_conn, write_capital_flow_cache
 
 DATA_DIR = "/home/hmo/web-dashboard/data"
 CACHE_PATH = f"{DATA_DIR}/capital_flow_cache.json"
@@ -53,7 +53,9 @@ def fetch_flow(code, days=5):
     data = _rate_limited_request(url)
     if not data:
         return None
-    klines = data.get("data", {}).get("klines", [])
+    klines = data.get("data") or {}
+    if isinstance(klines, dict):
+        klines = klines.get("klines", [])
     if not klines:
         return None
     result = []