diff --git a/osrt/data/ycbv.py b/osrt/data/ycbv.py
index 6f07c8042de4741b4a92777a5909c827a663559f..1c67c7b5309e2ff24e64f2e5bdb86dffb87e76d7 100644
--- a/osrt/data/ycbv.py
+++ b/osrt/data/ycbv.py
@@ -326,8 +326,8 @@ class YCBVideo2D(Dataset):
             path_synth = self.path + "/train_synt/"
             self.images = extract_images_path(path_real, mode, self.images)
             self.images = extract_images_path(path_synth, mode, self.images)
-            self.masks = extract_images_path(path_real, mode, self.masks, "masks")
-            self.masks = extract_images_path(path_synth, mode, self.masks, "masks")
+            self.masks = extract_images_path(path_real, mode, self.masks, "mask")
+            self.masks = extract_images_path(path_synth, mode, self.masks, "mask")
         dataset_name = 'YCB'
 
         print(f"Load dataset {dataset_name} in mode {self.mode}")