diff --git a/autogalaxy/aggregator/imaging/imaging.py b/autogalaxy/aggregator/imaging/imaging.py
index 8953b1e0..1660dccd 100644
--- a/autogalaxy/aggregator/imaging/imaging.py
+++ b/autogalaxy/aggregator/imaging/imaging.py
@@ -13,6 +13,8 @@
 - ``ImagingAgg`` — a class wrapping an ``Aggregator`` that returns a lazy generator of
   ``Imaging`` datasets, avoiding loading all results into memory at once.
 """
+
+from __future__ import annotations
 from functools import partial
 from typing import List
 
diff --git a/autogalaxy/aggregator/interferometer/interferometer.py b/autogalaxy/aggregator/interferometer/interferometer.py
index 3dc83bfa..6888f11e 100644
--- a/autogalaxy/aggregator/interferometer/interferometer.py
+++ b/autogalaxy/aggregator/interferometer/interferometer.py
@@ -15,6 +15,8 @@
   generator of ``Interferometer`` datasets, avoiding loading all results into memory at
   once.
 """
+
+from __future__ import annotations
 from functools import partial
 from typing import List