diff --git a/mu_map/dataset/default.py b/mu_map/dataset/default.py
index 6282f78e55b0e7bda9fd3daa6821a0438cf390f6..db22f357b25da801c77b5480e745da40063ad9cd 100644
--- a/mu_map/dataset/default.py
+++ b/mu_map/dataset/default.py
@@ -127,8 +127,8 @@ class MuMapDataset(Dataset):
             csv_file=os.path.basename(self.csv_file),
             split_file=os.path.basename(self.split_file),
             split_name=split_name,
-            images_dir=os.path.basename(self.images_dir),
-            bed_contours_file=self.bed_contours_file,
+            images_dir=os.path.basename(self.dir_images),
+            bed_contours_file=os.path.basename(self.bed_contours_file),
             discard_mu_map_slices=self.discard_mu_map_slices,
             align=self.align,
             scatter_correction=self.scatter_correction,