diff --git a/autogalaxy/__init__.py b/autogalaxy/__init__.py
index 652687577..ea6291da2 100644
--- a/autogalaxy/__init__.py
+++ b/autogalaxy/__init__.py
@@ -6,6 +6,9 @@
 
 from autoconf.dictable import from_dict, from_json, output_to_json, to_dict
 from autoarray.dataset import preprocess  # noqa
+from autoarray.dataset.interferometer.w_tilde import (
+    load_curvature_preload_if_compatible,
+)
 from autoarray.dataset.imaging.dataset import Imaging  # noqa
 from autoarray.dataset.interferometer.dataset import Interferometer  # noqa
 from autoarray.dataset.dataset_model import DatasetModel
diff --git a/autogalaxy/aggregator/interferometer/interferometer.py b/autogalaxy/aggregator/interferometer/interferometer.py
index 1886ad250..117264be4 100644
--- a/autogalaxy/aggregator/interferometer/interferometer.py
+++ b/autogalaxy/aggregator/interferometer/interferometer.py
@@ -53,7 +53,7 @@ def _interferometer_from(
         )
         uv_wavelengths = fit.value(name="dataset")[3].data
 
-        transformer_class = fit.value(name="dataset.transformer_class")
+        transformer_class = fit.value(name="transformer_class")
 
         dataset = aa.Interferometer(
             data=data,
diff --git a/autogalaxy/interferometer/model/analysis.py b/autogalaxy/interferometer/model/analysis.py
index fa009b871..ac4a56c41 100644
--- a/autogalaxy/interferometer/model/analysis.py
+++ b/autogalaxy/interferometer/model/analysis.py
@@ -185,5 +185,6 @@ def save_attributes(self, paths: af.DirectoryPaths):
         super().save_attributes(paths=paths)
 
         paths.save_json(
-            "transformer_class", to_dict(self.dataset.transformer.__class__), "dataset"
+            "transformer_class",
+            to_dict(self.dataset.transformer.__class__),
         )
diff --git a/autogalaxy/profiles/mass/total/isothermal.py b/autogalaxy/profiles/mass/total/isothermal.py
index 2d941affc..ba26bf671 100644
--- a/autogalaxy/profiles/mass/total/isothermal.py
+++ b/autogalaxy/profiles/mass/total/isothermal.py
@@ -76,9 +76,6 @@ def deflections_yx_2d_from(self, grid: aa.type.Grid2DLike, xp=np, **kwargs):
         """
         Calculate the deflection angles on a grid of (y,x) arc-second coordinates.
 
-        For coordinates (0.0, 0.0) the analytic calculation of the deflection angle gives a NaN. Therefore,
-        coordinates at (0.0, 0.0) are shifted slightly to (1.0e-8, 1.0e-8).
-
         Parameters
         ----------
         grid