diff --git a/python/setup.py b/python/setup.py
index a135a6c41101..937d682e3c85 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -164,7 +164,7 @@ def get_package_data_files():
                                'matplotlib'],
                       'extra_feature': ['tornado',
                                         'psutil',
-                                        'xgboost',
+                                        'xgboost==0.90',
                                         'mypy',
                                         'orderedset',
                                         'antlr4-python3-runtime']},