diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp
index 18baede90..ab906aefc 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
index 515a7f6a0..b27ab5e0f 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer.py
@@ -216,9 +216,10 @@ class Conformer(EncoderInterface):
                                 )
                             )
         '''
+        
         x = 0
         for enum, alpha in enumerate(self.alpha):
-            x += self.sigmoid(alpha)*layer_output[self.group_size*(enum+1)-1]
+            x += self.sigmoid(alpha)*layer_output[enum]
         
         x = self.layer_norm((1/self.group_size)*x)