diff --git a/examples/references/segmentation/pascal_voc2012/dataflow.py b/examples/references/segmentation/pascal_voc2012/dataflow.py
index 6faf0c5a8948..77a36e3c9809 100644
--- a/examples/references/segmentation/pascal_voc2012/dataflow.py
+++ b/examples/references/segmentation/pascal_voc2012/dataflow.py
@@ -206,7 +206,7 @@ def get_inference_dataloader(
     )
 
 
-def ignore_mask_boundaries(force_apply, **kwargs):
+def ignore_mask_boundaries(**kwargs):
     assert "mask" in kwargs, "Input should contain 'mask'"
     mask = kwargs["mask"]
     mask[mask == 255] = 0