diff --git a/setup.py b/setup.py
index 4f20792..022d55b 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 packages = [
-    'tensorflow==1.13.1',
+    'tensorflow==2.4.0',
     'numpy',  
     'gym',
     'stable_baselines==2.7.0',