diff --git a/egs/librispeech/ASR/zipformer/attention_decoder.py b/egs/librispeech/ASR/zipformer/attention_decoder.py
index 71be2d1ebb..81682e87b5 100644
--- a/egs/librispeech/ASR/zipformer/attention_decoder.py
+++ b/egs/librispeech/ASR/zipformer/attention_decoder.py
@@ -22,11 +22,11 @@
 import k2
 import torch
 import torch.nn as nn
-
 from label_smoothing import LabelSmoothingLoss
-from icefall.utils import add_eos, add_sos, make_pad_mask
 from scaling import penalize_abs_values_gt
 
+from icefall.utils import add_eos, add_sos, make_pad_mask
+
 
 class AttentionDecoderModel(nn.Module):
     """