diff --git a/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py b/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py
index 57401955..541d83e8 100644
--- a/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py
+++ b/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py
@@ -18,7 +18,7 @@ class XarmProcessor:
 
     def __init__(self, folder_path: str, fps: int | None = None):
         self.folder_path = Path(folder_path)
-        self.keys = {"actions", "rewards", "dones", "masks"}
+        self.keys = {"actions", "rewards", "dones"}
         self.nested_keys = {"observations": {"rgb", "state"}, "next_observations": {"rgb", "state"}}
         if fps is None:
             fps = 15