diff --git a/autogalaxy/aggregator/__init__.py b/autogalaxy/aggregator/__init__.py
index 56c0a6dc7..8ed775de6 100644
--- a/autogalaxy/aggregator/__init__.py
+++ b/autogalaxy/aggregator/__init__.py
@@ -8,3 +8,4 @@
 from autogalaxy.aggregator.ellipse.fit_ellipse import FitEllipseAgg
 from autogalaxy.aggregator.subplot import SubplotDataset as subplot_dataset
 from autogalaxy.aggregator.subplot import SubplotFit as subplot_fit
+from autogalaxy.aggregator.subplot import FITSFit as fits_fit
diff --git a/autogalaxy/aggregator/subplot.py b/autogalaxy/aggregator/subplot.py
index 72f23f11a..30aa03763 100644
--- a/autogalaxy/aggregator/subplot.py
+++ b/autogalaxy/aggregator/subplot.py
@@ -1,6 +1,17 @@
 from enum import Enum
 
 
+class FITSFit(Enum):
+    """
+    The HDUs that can be extracted from the fit.fits file.
+    """
+
+    model_image = "MODEL_IMAGE"
+    residual_map = "RESIDUAL_MAP"
+    normalized_residual_map = "NORMALIZED_RESIDUAL_MAP"
+    chi_squared_map = "CHI_SQUARED_MAP"
+
+
 class SubplotDataset(Enum):
     """
     The subplots that can be extracted from the subplot_fit image.
@@ -15,7 +26,7 @@ class SubplotDataset(Enum):
     psf_log_10 = (1, 1)
     signal_to_noise_map = (2, 1)
     over_sample_size_lp = (0, 2)
-    over_sample_Size_pixelization = (1, 2)
+    over_sample_size_pixelization = (1, 2)
 
 
 class SubplotFit(Enum):