diff --git a/requirements.txt b/requirements.txt
index a367be5..3efd366 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,7 +10,7 @@ gast==0.2.0
 grpcio==1.16.0
 gunicorn==19.9.0
 h5py==2.8.0
-Keras==2.2.2
+Keras==2.2.4
 Keras-Applications==1.0.6
 Keras-Preprocessing==1.0.5
 kiwisolver==1.0.1
diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py
index b00af5d..9b8c104 100644
--- a/style_transfer/style_transfer.py
+++ b/style_transfer/style_transfer.py
@@ -55,7 +55,7 @@ class StyleTransfer():
 
         frames = cls._resize_images(frames, size=384)
         frames = np.stack(frames)
-        frames = frames / frames.max()
+        frames = frames / 255
         stylized_imgs = comixGAN_model.generator.predict(frames)
         return list(255 * stylized_imgs)