diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
index dde0940c5..2edfe376e 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
@@ -355,8 +355,8 @@ class ConformerEncoder(nn.Module):
                 indexes[j+1] = a
 
 
-        for i,j in enumerate(indexes):
-            mod = self.layers[j]
+        for i in indexes:
+            mod = self.layers[i]
             output, attn_scores = mod(
                 output,
                 pos_emb,