diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py
index 35991f5e9..13ba99017 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py
@@ -506,8 +506,8 @@ def compute_loss(
         # overwhelming the simple_loss and causing it to diverge,
         # in case it had not fully learned the alignment yet.
         pruned_loss_scale = (0.0 if warmup < 1.0 else
-                             (0.1 if warmup > 1.0 and warmup < 2.0) else
-                             1.0)
+                             (0.1 if warmup > 1.0 and warmup < 2.0 else
+                             1.0))
         loss = (params.simple_loss_scale * simple_loss +
                 pruned_loss_scale * pruned_loss)