diff --git a/egs/libriheavy/LM/zipformer1/train.py b/egs/libriheavy/LM/zipformer1/train.py
index e3964f6bf..3fda65eb4 100755
--- a/egs/libriheavy/LM/zipformer1/train.py
+++ b/egs/libriheavy/LM/zipformer1/train.py
@@ -794,7 +794,7 @@ def train_one_epoch(
             display_and_save_batch(batch, params=params)
             raise
 
-        if params.print_diagnostics and batch_idx == 5:
+        if params.print_diagnostics and _batch_idx == 5:
             return
 
         if (