diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
index 0ded01285..69935f836 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
@@ -113,17 +113,18 @@ class Conformer(EncoderInterface):
         self.num_left_chunks = num_left_chunks
 
         encoder_layer = ConformerEncoderLayer(
-            d_model,
-            nhead,
-            dim_feedforward,
-            dropout,
-            layer_dropout,
-            cnn_module_kernel,
+            d_model=d_model,
+            nhead=nhead,
+            dim_feedforward=dim_feedforward,
+            dropout=dropout,
+            layer_dropout=layer_dropout,
+            cnn_module_kernel=cnn_module_kernel,
+            causal=causal,
         )
         # aux_layers from 1/3
         self.encoder = ConformerEncoder(
-            encoder_layer,
-            num_encoder_layers,
+            encoder_layer=encoder_layer,
+            num_encoder_layers=num_encoder_layers,
             aux_layers=list(
                 range(
                     num_encoder_layers // 3,