diff --git a/dan/predict/attention.py b/dan/predict/attention.py
index 117adb78cf0e76b6af85837d855f3ee420c9df50..4ef781c3475bf76b7d4b999afb0310ea58d3b27a 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -226,7 +226,6 @@ def get_polygon(text, max_value, offset, weights, size=None, return_contours=Fal
     :param text: Text piece selected with offset after splitting DAN prediction
     :param max_value: Maximum "attention intensity" for parts of a text piece, used for normalization
     :param offset: Offset value to get the relevant part of text piece
-    :param attentions: Attention weights of size (n_char, feature_height, feature_width)
     :param size: Target size (width, height) to resize the coverage vector
     :param return_contours: Return the contour of the current polygon (used for plotting)
     """
@@ -285,7 +284,7 @@ def plot_attention(
     :param outname: Name of the gif image
     :param word_separators: List of word separators
     :param line_separators: List of line separators
-    :param output_polygons: Whether to plot extracted polygons
+    :param display_polygons: Whether to plot extracted polygons
     """
 
     height, width, _ = image.shape