diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless2/train.py b/egs/wenetspeech/ASR/pruned_transducer_stateless2/train.py
index 7d88653ac..e9691d06c 100644
--- a/egs/wenetspeech/ASR/pruned_transducer_stateless2/train.py
+++ b/egs/wenetspeech/ASR/pruned_transducer_stateless2/train.py
@@ -1032,7 +1032,6 @@ def scan_pessimistic_batches_for_oom(
     graph_compiler: CharCtcTrainingGraphCompiler,
     params: AttributeDict,
 ):
-    return
     from lhotse.dataset import find_pessimistic_batches
 
     logging.info(