diff --git a/dap/algos/addmask.py b/dap/algos/addmask.py
index 41e8576..cdfaf3c 100644
--- a/dap/algos/addmask.py
+++ b/dap/algos/addmask.py
@@ -1,6 +1,6 @@
 #TODO: find a better way to handle this
 
-def calc_apply_additional_mask(detector, pixel_mask_pf):
+def calc_apply_additional_mask(results, detector, pixel_mask_pf):
     apply_additional_mask = results.get("apply_additional_mask", False)
     if not apply_additional_mask:
         return
diff --git a/dap/worker.py b/dap/worker.py
index 07fe014..4c701a8 100644
--- a/dap/worker.py
+++ b/dap/worker.py
@@ -160,7 +160,7 @@ def work(backend_address, accumulator_host, accumulator_port, visualisation_host
                 pixel_mask_pf = None
 
 
-        calc_apply_additional_mask(detector, pixel_mask_pf)
+        calc_apply_additional_mask(results, detector, pixel_mask_pf)
 
 
         if pixel_mask_corrected is not None: