diff --git a/src/sparseml/transformers/compression/compressors/base.py b/src/sparseml/transformers/compression/compressors/base.py
index 22166e4a7b3..2a1a37d9196 100644
--- a/src/sparseml/transformers/compression/compressors/base.py
+++ b/src/sparseml/transformers/compression/compressors/base.py
@@ -74,7 +74,7 @@ def overwrite_weights(self, model_path: str, model: Module):
         """
         Overwrites the weights in model with weights decompressed from model_path
 
-        :param pretrained_model_path: path to compressed weights
+        :param model_path: path to compressed weights
         :param model: pytorch model to load decompressed weights into
         """
         dense_gen = self.decompress(model_path)