diff --git a/solutionbox/ml_workbench/xgboost/setup.py b/solutionbox/ml_workbench/xgboost/setup.py
index 577e46d26..67f7a9dd9 100644
--- a/solutionbox/ml_workbench/xgboost/setup.py
+++ b/solutionbox/ml_workbench/xgboost/setup.py
@@ -27,7 +27,7 @@
   long_description="""
   """,
   install_requires=[
-    'tensorflow==1.15.2',
+    'tensorflow==2.7.2',
     'protobuf==3.4.0',
     'pillow==6.2.0',  # ML Engine does not have PIL installed
     'xgboost==0.6a2',