diff --git a/RoboGym_training/animalai_train/setup.py b/RoboGym_training/animalai_train/setup.py
index 1ef3f4e..2de372e 100644
--- a/RoboGym_training/animalai_train/setup.py
+++ b/RoboGym_training/animalai_train/setup.py
@@ -22,7 +22,7 @@
     install_requires=[
         'animalai>=1.0.5',
         'dopamine-rl',
-        'tensorflow==1.14',
+        'tensorflow==2.11.1',
         'matplotlib',
         'Pillow>=4.2.1,<=5.4.1',
         'numpy>=1.13.3,<=1.14.5',