diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 5827301..de2edab 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -249,7 +249,7 @@ class Model(pl.LightningModule):
         else:
             waveform = Inference.prepare_output(waveform, model_sampling_rate,
                                 audio, sampling_rate)    
-        return waveform
+            return waveform
     @property
     def valid_monitor(self):