diff --git a/darts-segmentation/src/darts_segmentation/training/data.py b/darts-segmentation/src/darts_segmentation/training/data.py
index 5e1698c..b416ad1 100644
--- a/darts-segmentation/src/darts_segmentation/training/data.py
+++ b/darts-segmentation/src/darts_segmentation/training/data.py
@@ -146,7 +146,7 @@ def setup(self, stage: Literal["fit", "validate", "test", "predict"] | None = No
         train_idx, val_idx = list(kf.split(dataset))[self.current_fold]
         self.train = Subset(dataset, train_idx)
         self.val = Subset(dataset, val_idx)
-        self.val.dataset.transform = False
+        self.val.dataset.transform = None
 
     def train_dataloader(self):
         return DataLoader(self.train, batch_size=self.batch_size, num_workers=self.num_workers, shuffle=True)