diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index 3058bf0..2f36004 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -611,7 +611,7 @@ class AfterDetailerScript(scripts.Script):
         with ChangeTorchLoad():
             pred = predictor(ad_model, pp.image, args.ad_conf, **kwargs)
 
-        masks = mask_preprocess(pred.masks)
+        masks = mask_preprocess(pred.masks, args.ad_dilate_erode)
 
         if not masks:
             print(