diff --git a/tortoise/utils/stft.py b/tortoise/utils/stft.py
index 8de6bfb..f54eb96 100644
--- a/tortoise/utils/stft.py
+++ b/tortoise/utils/stft.py
@@ -116,7 +116,7 @@ class STFT(torch.nn.Module):
             assert(filter_length >= win_length)
             # get window and zero center pad it to filter_length
             fft_window = get_window(window, win_length, fftbins=True)
-            fft_window = pad_center(fft_window, filter_length)
+            fft_window = pad_center(fft_window, size=filter_length)
             fft_window = torch.from_numpy(fft_window).float()
 
             # window the bases