diff --git a/dan/predict/attention.py b/dan/predict/attention.py
index 84ecfc394936fcb7daaeeef3c5e1b9fbfe3cefc4..c9182597dc18fc483cb5a81dbe069d2d1e42ce35 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -119,7 +119,7 @@ def get_predicted_polygons_with_confidence(
         polygon, _ = get_polygon(
             text_piece,
             max_value,
-            offset,
+            start_index,
             weights,
             threshold_method=threshold_method,
             threshold_value=threshold_value,