diff --git a/enhancer/models/dccrn.py b/enhancer/models/dccrn.py
index c6ee837..72d8a23 100644
--- a/enhancer/models/dccrn.py
+++ b/enhancer/models/dccrn.py
@@ -325,6 +325,6 @@ class DCCRN(Model):
             imag = imag * mask_imag
 
         spec = torch.cat([real, imag], 1)
-        wav = self.istft(spec).squeeze(1)
+        wav = self.istft(spec)
         wav = wav.clamp_(-1, 1)
         return wav