diff --git a/dan/ocr/decoder.py b/dan/ocr/decoder.py
index bba35b99270415a50194a4754056bfffa78a217c..496d9b2e357caa290a16b3c7adbd1f5a4924a26c 100644
--- a/dan/ocr/decoder.py
+++ b/dan/ocr/decoder.py
@@ -476,7 +476,7 @@ class CTCLanguageDecoder:
     :param tokens_path: Path to a file containing valid tokens. If using a file, the expected
             format is for tokens mapping to the same index to be on the same line.
     :param language_model_weight: Weight of the language model.
-    :param temperature_scaling: Temperature for model calibreation.
+    :param temperature: Temperature for model calibreation.
     """
 
     def __init__(
@@ -495,7 +495,7 @@ class CTCLanguageDecoder:
         }
         self.index_to_token = {i: token for token, i in self.tokens_to_index.items()}
         self.blank_token_id = self.tokens_to_index[self.mapping.ctc.encoded]
-
+        
         # Torchaudio's decoder
         # https://pytorch.org/audio/master/generated/torchaudio.models.decoder.ctc_decoder.html
         self.decoder = ctc_decoder(