diff --git a/cnn_classification_service/cnn_classifier.py b/cnn_classification_service/cnn_classifier.py
index ec61935..df8570b 100644
--- a/cnn_classification_service/cnn_classifier.py
+++ b/cnn_classification_service/cnn_classifier.py
@@ -22,7 +22,7 @@ class Classifier(object):
 
         self.loaded_model.load_weights(weights_filename)
         self.datagen = ImageDataGenerator(rescale=1. / 255., validation_split=0.25)
-        self.loaded_model.compile(optimizers.rmsprop(lr=0.0005, decay=1e-6), loss="categorical_crossentropy",
+        self.loaded_model.compile(optimizers.RMSprop(lr=0.0005, decay=1e-6), loss="categorical_crossentropy",
                                   metrics=["accuracy"])
         self.loaded_model.summary()