diff --git a/src/correct_images/corrector.py b/src/correct_images/corrector.py
index f7abbc41..731e148a 100644
--- a/src/correct_images/corrector.py
+++ b/src/correct_images/corrector.py
@@ -811,12 +811,14 @@ def generate_attenuation_correction_parameters(self):
                     self.image_channels,
                 ),
                 dtype=np.float32,
+                memmap_location=self.memmap_location,
             )
             self.memmaps_to_remove.append(images_fn)
 
             distances_fn, distances_map = open_memmap(
                 shape=(len(hist_bins) - 1, self.image_height * self.image_width),
                 dtype=np.float32,
+                memmap_location=self.memmap_location,
             )
             self.memmaps_to_remove.append(distances_fn)
 
@@ -923,6 +925,7 @@ def generate_attenuation_correction_parameters(self):
                     self.image_channels,
                 ),
                 dtype=np.float32,
+                memmap_location=self.memmap_location,
             )
             """