diff --git a/utils/lgb_predict_task2.py b/utils/lgb_predict_task2.py
index 3647a1267f8f744cbb635ca22bcf0ae29d8edfc7..c899e3926074ce048b95de232f8bb75cdff3d80f 100755
--- a/utils/lgb_predict_task2.py
+++ b/utils/lgb_predict_task2.py
@@ -7,21 +7,21 @@ __MAP__ = ["irrelevant", "complement", "substitute", "exact"]
 LGB_CONFIG = {
     "us": {
         "product_feat": "/models/us-product-feat.csv",
-        "model_file": "/models/lgb-us.txt",
+        "model_file": "/models/lgb-us-task-2.txt",
         "features": ['pred_0','pred_1','pred_2','pred_3',
             'label_0','label_1','label_2','label_3',
             'query_count','is_isbn','has_isbn','fold']
     },
     "jp": {
         "product_feat": "/models/jp-product-feat.csv",
-        "model_file": "/models/lgb-es-jp.txt",
+        "model_file": "/models/lgb-es-jp-task-2.txt",
         "features": ['pred_0','pred_1','pred_2','pred_3',
             'label_0','label_1','label_2','label_3',
             'query_count','is_isbn','has_isbn','fold','locale']
     },
     "es": {
         "product_feat": "/models/es-product-feat.csv",
-        "model_file": "/models/lgb-es-jp.txt",
+        "model_file": "/models/lgb-es-jp-task-2.txt",
         "features": ['pred_0','pred_1','pred_2','pred_3',
             'label_0','label_1','label_2','label_3',
             'query_count','is_isbn','has_isbn','fold','locale']