diff --git a/spleeter/audio/adapter.py b/spleeter/audio/adapter.py
index bda1441..720618c 100644
--- a/spleeter/audio/adapter.py
+++ b/spleeter/audio/adapter.py
@@ -13,7 +13,7 @@ from os.path import exists
 import numpy as np
 import tensorflow as tf
 
-from tensorflow.contrib.signal import stft, hann_window
+from tensorflow.signal import stft, hann_window
 # pylint: enable=import-error
 
 from .. import SpleeterError
diff --git a/spleeter/audio/spectrogram.py b/spleeter/audio/spectrogram.py
index 38d0cf5..f16782d 100644
--- a/spleeter/audio/spectrogram.py
+++ b/spleeter/audio/spectrogram.py
@@ -7,7 +7,7 @@
 import numpy as np
 import tensorflow as tf
 
-from tensorflow.contrib.signal import stft, hann_window
+from tensorflow.signal import stft, hann_window
 # pylint: enable=import-error
 
 __email__ = 'research@deezer.com'