diff --git a/autolens/util/__init__.py b/autolens/util/__init__.py
index b4759c327..10cc4fc32 100644
--- a/autolens/util/__init__.py
+++ b/autolens/util/__init__.py
@@ -18,6 +18,7 @@
     inversion_interferometer_util as inversion_interferometer,
 )
 from autoarray.operators import transformer_util as transformer
+from autoarray.util import dataset_util as dataset
 from autogalaxy.util import error_util as error
 from autogalaxy.analysis import chaining_util as chaining