diff --git a/parakeet/modules/stft_loss.py b/parakeet/modules/stft_loss.py
index 1f400b46..8af55ab1 100644
--- a/parakeet/modules/stft_loss.py
+++ b/parakeet/modules/stft_loss.py
@@ -51,7 +51,7 @@ def stft(x,
     # calculate window
     window = signal.get_window(window, win_length, fftbins=True)
     window = paddle.to_tensor(window)
-    x_stft = paddle.tensor.signal.stft(
+    x_stft = paddle.signal.stft(
         x,
         fft_size,
         hop_length,