diff --git a/dan/ocr/decoder.py b/dan/ocr/decoder.py
index 644f8d3b5692ec71c1f51b2c3328e2c86d55c3bd..b62c0f6c77b48518dd0939e984c186b1377e887d 100644
--- a/dan/ocr/decoder.py
+++ b/dan/ocr/decoder.py
@@ -597,8 +597,8 @@ class CTCLanguageDecoder:
     ) -> Dict[str, List[Union[str, float]]]:
         """
         Decode a feature vector using n-gram language modelling.
-        :param features: Feature vector of size (batch_size, n_tokens, n_frames).
-        :param batch_sizes: Prediction length of size batch_size.
+        :param batch_features: Feature vector of size (batch_size, n_tokens, n_frames).
+        :param batch_frames: Prediction length of size batch_size.
         :return: A dictionary containing the hypotheses and their confidences.
         """
         # Reshape from (batch_size, n_tokens, n_frames) to (batch_size, n_frames, n_tokens)
diff --git a/docs/get_started/training.md b/docs/get_started/training.md
index 92eda24c834fb83765a07aeff5f5242a0be34d59..1cb5a0cffb55d6aaf9c9e2fee9fc65e433add1af 100644
--- a/docs/get_started/training.md
+++ b/docs/get_started/training.md
@@ -32,4 +32,4 @@ To train a DAN model, please refer to the [documentation of the training command
 
 ## 3. Predict
 
-Once the training is complete, you can apply  a trained DAN model on an image using the [predict command](../usage/predict.md) and the `inference_parameters.yml` file, located in `{training.output_folder}/results`.
+Once the training is complete, you can apply  a trained DAN model on an image using the [predict command](../usage/predict/index.md) and the `inference_parameters.yml` file, located in `{training.output_folder}/results`.
diff --git a/docs/usage/index.md b/docs/usage/index.md
index c926d9b9c846c23b85d74ac8d0b420f4cd399d35..1bd57fecfc4cf183748ca260f67d4767cb8fb932 100644
--- a/docs/usage/index.md
+++ b/docs/usage/index.md
@@ -9,4 +9,4 @@ When `teklia-dan` is installed in your environment, you may use the following co
 : To train a new DAN model. More details in [the dedicated section](./train/index.md).
 
 `teklia-dan predict`
-: To predict an image using a trained DAN model. More details in [the dedicated section](./predict.md).
+: To predict an image using a trained DAN model. More details in [the dedicated section](./predict/index.md).