diff --git a/python/source/setup.py b/python/source/setup.py
index bc89975..e548c3b 100644
--- a/python/source/setup.py
+++ b/python/source/setup.py
@@ -30,7 +30,7 @@
             'Programming Language :: Python :: 3.6'
         ],
         install_requires=[
-            'keras==2.1.6',
+            'keras==2.13.1',
             'scikit-image==0.14.0',
             'imageio==2.4.1',
             'jupyterlab',