diff --git a/autoarray/dataset/imaging/dataset.py b/autoarray/dataset/imaging/dataset.py
index 85bc3faa..c464563d 100644
--- a/autoarray/dataset/imaging/dataset.py
+++ b/autoarray/dataset/imaging/dataset.py
@@ -381,7 +381,7 @@ def apply_noise_scaling(self, mask: Mask2D, noise_value: float = 1e8) -> "Imagin
             noise_covariance_matrix=self.noise_covariance_matrix,
             over_sampling=self.over_sampling,
             pad_for_convolver=False,
-            check_noise_map=False
+            check_noise_map=False,
         )
 
         logger.info(
@@ -434,7 +434,7 @@ def apply_over_sampling(
             psf=self.psf,
             over_sampling=over_sampling,
             pad_for_convolver=False,
-            check_noise_map=False
+            check_noise_map=False,
         )
 
     def output_to_fits(
diff --git a/autoarray/plot/abstract_plotters.py b/autoarray/plot/abstract_plotters.py
index d179533d..07db0729 100644
--- a/autoarray/plot/abstract_plotters.py
+++ b/autoarray/plot/abstract_plotters.py
@@ -37,7 +37,6 @@ def __init__(
 
         self.subplot_figsize = None
 
-
     def set_title(self, label):
         if self.mat_plot_1d is not None:
             self.mat_plot_1d.title.manual_label = label
diff --git a/autoarray/plot/wrap/base/figure.py b/autoarray/plot/wrap/base/figure.py
index 51fe28a1..7e38cc85 100644
--- a/autoarray/plot/wrap/base/figure.py
+++ b/autoarray/plot/wrap/base/figure.py
@@ -94,4 +94,5 @@ def close(self):
         """
         plt.close()
         import gc
+
         gc.collect()
diff --git a/autoarray/preloads.py b/autoarray/preloads.py
index bb9c7d28..6808f0f6 100644
--- a/autoarray/preloads.py
+++ b/autoarray/preloads.py
@@ -325,6 +325,9 @@ def set_linear_func_inversion_dicts(self, fit_0, fit_1):
                 < 1e-8
             ):
                 should_preload = True
+            else:
+                should_preload = False
+                break
 
         if should_preload:
             self.linear_func_operated_mapping_matrix_dict = (