diff --git a/autolens/aggregator/fit_imaging.py b/autolens/aggregator/fit_imaging.py
index 855cade15..a97954b12 100644
--- a/autolens/aggregator/fit_imaging.py
+++ b/autolens/aggregator/fit_imaging.py
@@ -14,6 +14,8 @@
   that exposes a generator of ``FitImaging`` objects, enabling memory-efficient iteration
   over large result sets.
 """
+
+from __future__ import annotations
 from typing import Optional, List
 
 import autofit as af
diff --git a/autolens/aggregator/fit_interferometer.py b/autolens/aggregator/fit_interferometer.py
index ac6389050..b23ba7bea 100644
--- a/autolens/aggregator/fit_interferometer.py
+++ b/autolens/aggregator/fit_interferometer.py
@@ -15,6 +15,8 @@
   ``Aggregator`` that exposes a generator of ``FitInterferometer`` objects, enabling
   memory-efficient iteration over large result sets.
 """
+
+from __future__ import annotations
 from typing import Optional, List
 
 import autofit as af
diff --git a/autolens/aggregator/subhalo.py b/autolens/aggregator/subhalo.py
index 1e361c3da..0a603181d 100644
--- a/autolens/aggregator/subhalo.py
+++ b/autolens/aggregator/subhalo.py
@@ -10,6 +10,8 @@
 for the best-fit model in each cell — both the "with subhalo" fit and (optionally) the
 "no subhalo" baseline — enabling the per-cell log-evidence difference map to be computed.
 """
+
+from __future__ import annotations
 from typing import Optional
 
 import autofit as af