diff --git a/denoisers/__init__.py b/denoisers/__init__.py
index 3f891b9..b7347d9 100644
--- a/denoisers/__init__.py
+++ b/denoisers/__init__.py
@@ -1,5 +1,5 @@
 """Denoisers for the 1D and 2D cases."""
-from modeling.waveunet.config import WaveUNetConfig
-from modeling.waveunet.model import WaveUNetModel
+from denoisers.modeling.waveunet.config import WaveUNetConfig
+from denoisers.modeling.waveunet.model import WaveUNetModel
 
 __all__ = ["WaveUNetConfig", "WaveUNetModel"]
diff --git a/denoisers/modeling/__init__.py b/denoisers/modeling/__init__.py
index 67f0447..d942cd3 100644
--- a/denoisers/modeling/__init__.py
+++ b/denoisers/modeling/__init__.py
@@ -1 +1,5 @@
 """Models."""
+from denoisers.modeling.waveunet.config import WaveUNetConfig
+from denoisers.modeling.waveunet.model import WaveUNetModel
+
+__all__ = ["WaveUNetConfig", "WaveUNetModel"]
diff --git a/denoisers/modeling/waveunet/__init__.py b/denoisers/modeling/waveunet/__init__.py
index 2ea9cd6..23b1ac8 100644
--- a/denoisers/modeling/waveunet/__init__.py
+++ b/denoisers/modeling/waveunet/__init__.py
@@ -1 +1,5 @@
 """1d UNet model."""
+from denoisers.modeling.waveunet.config import WaveUNetConfig
+from denoisers.modeling.waveunet.model import WaveUNetModel
+
+__all__ = ["WaveUNetConfig", "WaveUNetModel"]