diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
index 2668a2f15..7f6616eff 100644
Binary files a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp differ
diff --git a/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp b/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp
index b1ef9a8ab..9e8ab3ca9 100644
Binary files a/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.transformer.py.swp differ
diff --git a/egs/librispeech/ASR/conformer_ctc2/transformer.py b/egs/librispeech/ASR/conformer_ctc2/transformer.py
index e7b1c3b37..737bd9e05 100644
--- a/egs/librispeech/ASR/conformer_ctc2/transformer.py
+++ b/egs/librispeech/ASR/conformer_ctc2/transformer.py
@@ -263,6 +263,7 @@ class Transformer(nn.Module):
         sos_id: int,
         eos_id: int,
         warmup: float = 1.0,
+        return_output: bool = False,
     ) -> torch.Tensor:
         """
         Args: