diff --git a/enhancer/models/__init__.py b/enhancer/models/__init__.py
index dd12bc3..0d21337 100644
--- a/enhancer/models/__init__.py
+++ b/enhancer/models/__init__.py
@@ -1 +1,2 @@
-from enhancer.models.demucs import Demucs
\ No newline at end of file
+from enhancer.models.demucs import Demucs
+from enhancer.models.waveunet import WaveUnet
\ No newline at end of file