diff --git a/src/mednet/libs/common/data/datamodule.py b/src/mednet/libs/common/data/datamodule.py
index e6d0ca0e6dd38cbf6b4c1ada161294e57c7f2d02..3bfcb9f52846db648ef71b37835330e225d69d9c 100644
--- a/src/mednet/libs/common/data/datamodule.py
+++ b/src/mednet/libs/common/data/datamodule.py
@@ -570,7 +570,14 @@ class ConcatDataModule(lightning.LightningDataModule):
             The list of validation dataset names.
         """
 
-        return ["validation"] + [
+        validation_split_name = "validation"
+        if "validation" not in self.splits.keys():
+            logger.warning(
+                "No split named 'validation', the training split will be used for validation instead."
+            )
+            validation_split_name = "train"
+
+        return [validation_split_name] + [
             k for k in self.splits.keys() if k.startswith("monitor-")
         ]