diff --git a/cnn_classification_service/cnn_classifier.py b/cnn_classification_service/cnn_classifier.py
index 041b7ad..617527d 100644
--- a/cnn_classification_service/cnn_classifier.py
+++ b/cnn_classification_service/cnn_classifier.py
@@ -10,9 +10,9 @@ import librosa
 import librosa.display
 import numpy
 import matplotlib.pyplot
-from keras.models import model_from_json
-from keras import optimizers
-from keras_preprocessing.image import ImageDataGenerator
+from tensorflow.keras.models import model_from_json
+from tensorflow.keras import optimizers
+from tensorflow.keras_preprocessing.image import ImageDataGenerator
 
 
 class Classifier(object):