diff --git a/dh_segment_torch/post_processing/geometries/polygon.py b/dh_segment_torch/post_processing/geometries/polygon.py
index 07c3aec..db94713 100644
--- a/dh_segment_torch/post_processing/geometries/polygon.py
+++ b/dh_segment_torch/post_processing/geometries/polygon.py
@@ -44,4 +44,4 @@ def apply(
         found_polygons = sorted(
             found_polygons, key=lambda poly: poly.area, reverse=True
         )
-        return found_polygons[: min(self.max_boxes, len(found_polygons))]
+        return found_polygons[: min(self.max_polygons, len(found_polygons))]