diff --git a/sanguo_data/index_downloader.py b/sanguo_data/index_downloader.py
index 691b13c..d4b24b2 100644
--- a/sanguo_data/index_downloader.py
+++ b/sanguo_data/index_downloader.py
@@ -24,7 +24,7 @@ def download_index(symbol: str, start_year: int, end_year: int, out_dir: str) ->
 
     # 登录 baostock
     lg = bs.login()
-    if lg.error_code != "success":
+    if lg.error_code not in ("0", "success"):
         raise RuntimeError(f"baostock 登录失败: {lg.error_msg}")
 
     try:
@@ -44,13 +44,13 @@ def download_index(symbol: str, start_year: int, end_year: int, out_dir: str) ->
                 adjustfield="3"  # 不复权
             )
 
-            if rs.error_code != "success":
+            if rs.error_code not in ("0", "success"):
                 print(f"Warning: 下载 {symbol} {year} 失败: {rs.error_msg}")
                 continue
 
             # 转换为 DataFrame
             data_list = []
-            while (rs.error_code == "success") and rs.next():
+            while (rs.error_code in ("0", "success")) and rs.next():
                 data_list.append(rs.get_row_data())
 
             if not data_list: