diff --git a/tools/train/train/dataset.py b/tools/train/train/dataset.py
index 0562eec..c558ef4 100644
--- a/tools/train/train/dataset.py
+++ b/tools/train/train/dataset.py
@@ -120,12 +120,12 @@ class Dataset(torch.utils.data.Dataset):
         # Check system prompt hash
         file_system_hash = root.get('system_prompt_hash')
         if file_system_hash != self.system_prompt_hash:
-            print(f"WARNING: System prompt hash mismatch in {file_path}")
+            print(f"WARNING: System prompt hash mismatch in {file}")
         
         # Check action schema hash
         file_schema_hash = root.get('action_schema_hash')
         if file_schema_hash != self.action_schema_hash:
-            print(f"WARNING: Action schema hash mismatch in {file_path}")
+            print(f"WARNING: Action schema hash mismatch in {file}")
         
         # Check for required elements
         context_elem = root.find('context')