diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
index 69935f836..9d63cb123 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
@@ -124,7 +124,7 @@ class Conformer(EncoderInterface):
         # aux_layers from 1/3
         self.encoder = ConformerEncoder(
             encoder_layer=encoder_layer,
-            num_encoder_layers=num_encoder_layers,
+            num_layers=num_encoder_layers,
             aux_layers=list(
                 range(
                     num_encoder_layers // 3,