diff --git a/asteroid/dsp/overlap_add.py b/asteroid/dsp/overlap_add.py
index 256b43bc5..c858e6b4e 100644
--- a/asteroid/dsp/overlap_add.py
+++ b/asteroid/dsp/overlap_add.py
@@ -113,7 +113,7 @@ def ola_forward(self, x):
                 frame = _reorder_sources(frame, out[-1], n_src, self.window_size, self.hop_size)
 
             if self.use_window:
-                frame = frame * self.window
+                frame = frame * self.window.to(frame)
             else:
                 frame = frame / (self.window_size / self.hop_size)
             out.append(frame)