diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml
index 8d0ab14..c0b2cf6 100644
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@@ -1,5 +1,5 @@
 defaults:
-  - model : Demucs
+  - model : WaveUnet
   - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default