diff --git a/egs/librispeech/ASR/conformer_lm/conformer.py b/egs/librispeech/ASR/conformer_lm/conformer.py
index fe0a5eec9..d08b64ea3 100644
--- a/egs/librispeech/ASR/conformer_lm/conformer.py
+++ b/egs/librispeech/ASR/conformer_lm/conformer.py
@@ -112,7 +112,7 @@ class MaskedLmConformer(nn.Module):
 
         Returns:
           Returns (memory, pos_emb), where:
-            `memory` is a Tensor containing the encoded data; it is of shape (N, T, C)
+            `memory` is a Tensor containing the encoded data; it is of shape (T, N, C)
                 where C is the embedding_dim.
             `pos_emb` is a Tensor containing the relative positional encoding, of
                       shape (1, 2*T-1, C)