diff --git a/finetune_kopa.py b/finetune_kopa.py
index 1487459..f5ad0ac 100644
--- a/finetune_kopa.py
+++ b/finetune_kopa.py
@@ -179,7 +179,7 @@ def train(
     slama_model = KoPAWithAdapter(model, num_prefix, kge_model=kge_model)
 
     if data_path.endswith(".json") or data_path.endswith(".jsonl"):
-        data = load_dataset("json")
+        data = load_dataset("json", data_files=data_path)
     else:
         data = load_dataset(data_path)