diff --git a/training/xgboost/structured/base/setup.py b/training/xgboost/structured/base/setup.py
index affdcc26f..302dc59c6 100644
--- a/training/xgboost/structured/base/setup.py
+++ b/training/xgboost/structured/base/setup.py
@@ -27,7 +27,7 @@
     'tensorflow==1.15.4',
     'scikit-learn==0.20.2',
     'pandas==0.24.2',
-    'xgboost==0.81',
+    'xgboost==1.4.2',
     'cloudml-hypertune',
 ]