diff --git a/keras-ai-platform/setup.py b/keras-ai-platform/setup.py
index cd14aa1..2b5d966 100644
--- a/keras-ai-platform/setup.py
+++ b/keras-ai-platform/setup.py
@@ -6,7 +6,7 @@
 NAME = 'trainer'
 VERSION = '1.0'
 REQUIRED_PACKAGES = [
-    'tensorflow==2.1.0',
+    'tensorflow==2.11.1',
     'pandas'
 ]