diff --git a/audiocraft/models/lm.py b/audiocraft/models/lm.py
index 8cefd2c5..75790e7d 100644
--- a/audiocraft/models/lm.py
+++ b/audiocraft/models/lm.py
@@ -225,7 +225,7 @@ def forward(self, sequence: torch.Tensor,
         S the sequence steps, return the logits with shape [B, card, K, S].
 
         Args:
-            indices (torch.Tensor): Indices of the codes to model.
+            sequence (torch.Tensor): Indices of the codes to model.
             conditions (list of ConditioningAttributes): Conditions to use when modeling
                 the given codes. Note that when evaluating multiple time with the same conditioning
                 you should pre-compute those and pass them as `condition_tensors`.