diff --git a/molecules/requirements.txt b/molecules/requirements.txt
index 6fc994391..9d470765c 100644
--- a/molecules/requirements.txt
+++ b/molecules/requirements.txt
@@ -1,3 +1,3 @@
 apache-beam[gcp]==2.5
 tensorflow-transform==0.8
-tensorflow==1.8
+tensorflow==2.5.1
diff --git a/molecules/setup.py b/molecules/setup.py
index b554f90f6..1edab6e56 100644
--- a/molecules/setup.py
+++ b/molecules/setup.py
@@ -18,7 +18,7 @@
 REQUIRED_PACKAGES = [
     'apache-beam[gcp]==2.5',
     'tensorflow-transform==0.8',
-    'tensorflow==1.8',
+    'tensorflow==2.5.1',
 ]
 
 setuptools.setup(