diff --git a/src/mednet/data/datamodule.py b/src/mednet/data/datamodule.py
index d7b6e93b6e79f51d4c545668ecb7d789e0f5b352..56ab3000a712c99f857f771ab36ecf85bebb6902 100644
--- a/src/mednet/data/datamodule.py
+++ b/src/mednet/data/datamodule.py
@@ -792,7 +792,7 @@ class ConcatDataModule(lightning.LightningDataModule):
         return torch.utils.data.DataLoader(
             self._datasets["train"],
             shuffle=(self._train_sampler is None),
-            batch_size=self.batch,
+            batch_size=self.batch_size,
             drop_last=self.drop_incomplete_batch,
             pin_memory=self.pin_memory,
             sampler=self._train_sampler,