diff --git a/iris_pipeline/__init__.py b/iris_pipeline/__init__.py
index 53e5bbc..4016e87 100644
--- a/iris_pipeline/__init__.py
+++ b/iris_pipeline/__init__.py
@@ -39,4 +39,4 @@ class UnsupportedPythonError(Exception):
 from .assign_wcs import *
 from .resample import *
 
-from .datamodels import  monkeypatch_jwst_datamodels
+from .datamodels import  *
diff --git a/iris_pipeline/datamodels/__init__.py b/iris_pipeline/datamodels/__init__.py
index c8a61c6..8e0977c 100644
--- a/iris_pipeline/datamodels/__init__.py
+++ b/iris_pipeline/datamodels/__init__.py
@@ -14,6 +14,7 @@
 
 
 __all__ = [
+    "monkeypatch_jwst_datamodels",
     "IRISImageModel",
     "TMTRampModel",
     "TMTMaskModel",
@@ -25,7 +26,7 @@
     "TMTReferenceFileModel",
 ]
 
-_all_models = __all__
+_all_models = __all__[1:]
 _local_dict = locals()
 _defined_models = {k: _local_dict[k] for k in _all_models}