diff --git a/spleeter/model/__init__.py b/spleeter/model/__init__.py
index 531f84c..20769d1 100644
--- a/spleeter/model/__init__.py
+++ b/spleeter/model/__init__.py
@@ -8,7 +8,7 @@ import importlib
 # pylint: disable=import-error
 import tensorflow as tf
 
-from tensorflow.contrib.signal import stft, inverse_stft, hann_window
+from tensorflow.signal import stft, inverse_stft, hann_window
 # pylint: enable=import-error
 
 from ..utils.tensor import pad_and_partition, pad_and_reshape