diff --git a/torchaudio/augmentations.py b/torchaudio/augmentations.py
index 52c6e2091b..efa07e9fd8 100644
--- a/torchaudio/augmentations.py
+++ b/torchaudio/augmentations.py
@@ -41,7 +41,7 @@ def forward(self, complex_specgrams, overriding_rate=None):
                 If no rate is passed, use ``self.fixed_rate``
 
         Returns:
-            (Tensor): Stretched complex spectrogram of dimension (*, channel, n_freq, ceil(time/rate), complex=2)
+            (Tensor): Stretched complex spectrogram of dimension (*, channel, freq, ceil(time/rate), complex=2)
         """
         assert complex_specgrams.size(-1) == 2, "complex_specgrams should be a complex tensor, shape (*, complex=2)"