diff --git a/solutionbox/ml_workbench/tensorflow/setup.py b/solutionbox/ml_workbench/tensorflow/setup.py
index a3c660af3..3d50309f6 100644
--- a/solutionbox/ml_workbench/tensorflow/setup.py
+++ b/solutionbox/ml_workbench/tensorflow/setup.py
@@ -27,7 +27,7 @@
   long_description="""
   """,
   install_requires=[
-    'tensorflow==1.15.2',
+    'tensorflow==2.6.4',
     'protobuf==3.1.0',
     'pillow==6.2.0',  # ML Engine does not have PIL installed
   ],