diff --git a/autogalaxy/util/__init__.py b/autogalaxy/util/__init__.py
index 32b7b7fa..827094a6 100644
--- a/autogalaxy/util/__init__.py
+++ b/autogalaxy/util/__init__.py
@@ -19,6 +19,7 @@
 )
 from autoarray.operators import transformer_util as transformer
 from autoarray.util import misc_util as misc
+from autoarray.util import dataset_util as dataset
 
 from autogalaxy.util import error_util as error
 from autogalaxy.analysis import chaining_util as chaining