diff --git a/birefnetNode.py b/birefnetNode.py
index cfafaf8..649511f 100644
--- a/birefnetNode.py
+++ b/birefnetNode.py
@@ -298,7 +298,12 @@ def get_foreground(self, images, masks, blur_size=91, blur_size_two=7, fill_colo
         _image_maskeds = []
         # for _image, _out_mask in images, out_masks:
         for idx, (_image, _out_mask) in enumerate(zip(images.unbind(dim=0), out_masks.unbind(dim=0))):
-            _image_masked = refine_foreground_pil(tensor_to_pil(_image), tensor_to_pil(_out_mask.permute(1, 2, 0)))
+            _image_masked = refine_foreground_pil(
+                tensor_to_pil(_image),
+                tensor_to_pil(_out_mask.permute(1, 2, 0)),
+                r1=blur_size,
+                r2=blur_size_two,
+            )
             _image_masked = pil_to_tensor(_image_masked)
             _image_maskeds.append(_image_masked)
             del _image_masked