diff --git a/ai-platform/named-entity-recognition-trainer/setup.py b/ai-platform/named-entity-recognition-trainer/setup.py
index 3f8f0bb..7a8029f 100644
--- a/ai-platform/named-entity-recognition-trainer/setup.py
+++ b/ai-platform/named-entity-recognition-trainer/setup.py
@@ -2,7 +2,7 @@
 from setuptools import setup
 
 REQUIRED_PACKAGES = [
-    'Keras==2.2.4',
+    'Keras==2.13.1',
     'cloudml-hypertune'
 ]