diff --git a/lerobot/configs/train.py b/lerobot/configs/train.py
index 7de0f879..d6e82438 100644
--- a/lerobot/configs/train.py
+++ b/lerobot/configs/train.py
@@ -106,9 +106,8 @@ class TrainPipelineConfig(HubMixin):
             train_dir = f"{now:%Y-%m-%d}/{now:%H-%M-%S}_{self.job_name}"
             self.output_dir = Path("outputs/train") / train_dir
 
-        if self.dataset is not None:
-            if isinstance(self.dataset.repo_id, list):
-                raise NotImplementedError("LeRobotMultiDataset is not currently implemented.")
+        if self.dataset is not None and isinstance(self.dataset.repo_id, list):
+            raise NotImplementedError("LeRobotMultiDataset is not currently implemented.")
 
         if not self.use_policy_training_preset and (self.optimizer is None or self.scheduler is None):
             raise ValueError("Optimizer and Scheduler must be set when the policy presets are not used.")