diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py
index 0e9fe1423..72cec1328 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py
@@ -861,7 +861,7 @@ def train_one_epoch(
                              params=params,
                              optimizer=optimizer,
                              scheduler=scheduler,
-                             sampler=sampler,
+                             sampler=train_dl.sampler,
                              scaler=scaler,
                              rank=0)