diff --git a/src/nanotron/trainer.py b/src/nanotron/trainer.py
index 0f6539c9..32b52169 100644
--- a/src/nanotron/trainer.py
+++ b/src/nanotron/trainer.py
@@ -335,7 +335,7 @@ def clear_dataloader_from_memory(dataloader: DataLoader, stage_name: str):
                         clear_dataloader_from_memory(prev_dataloader, stage_name=stage.name)
 
                 log_rank(
-                    f"[Training Stage: {stage.name}] Switching to a new dataset {stage.data.dataset.hf_dataset_or_datasets}",
+                    f"[Training Stage: {stage.name}] Switching to a new dataset",
                     logger=logger,
                     level=logging.INFO,
                     rank=0,