diff --git a/autolens/__init__.py b/autolens/__init__.py
index 91eb25351..b74a3cd85 100644
--- a/autolens/__init__.py
+++ b/autolens/__init__.py
@@ -1,6 +1,7 @@
 from autoconf import jax_wrapper
 from autoconf.dictable import from_dict, from_json, output_to_json, to_dict
 from autoarray import preprocess
+from autoarray.dataset.interferometer.w_tilde import load_curvature_preload_if_compatible
 from autoarray.dataset.imaging.w_tilde import WTildeImaging
 from autoarray.dataset.imaging.dataset import Imaging
 from autoarray.dataset.interferometer.dataset import (