diff --git a/examples/animalai_train/setup.py b/examples/animalai_train/setup.py
index 9d9bd20..0e08d39 100644
--- a/examples/animalai_train/setup.py
+++ b/examples/animalai_train/setup.py
@@ -22,7 +22,7 @@
     install_requires=[
         'animalai>=1.0.5',
         'dopamine-rl',
-        'tensorflow==1.14',
+        'tensorflow==2.9.3',
         'matplotlib',
         'Pillow>=4.2.1,<=5.4.1',
         'numpy>=1.13.3,<=1.14.5',