diff --git a/.travis.yml b/.travis.yml
index e02677915..36e49aaa7 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -389,7 +389,6 @@ matrix:
         # - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=-tf,-pytorch,-py37 python/ray/util/sgd/...
         - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=tf,-pytorch,-py37 python/ray/util/sgd/...
         - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=-tf,pytorch,-py37 python/ray/util/sgd/...
-        # - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only python/ray/util/xgboost/...
 
     # Docs: Tests and examples.
     - os: linux
@@ -418,6 +417,7 @@ matrix:
         - . ./ci/travis/ci.sh build
       script:
         - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=py37 python/ray/tune/...
+        - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only python/ray/util/xgboost/...
         # There are no python 3.7 tests for RaySGD at the moment
         # - ./ci/keep_alive bazel test --config=ci --build_tests_only --test_tag_filters=py37 python/ray/util/sgd/...
         # - ./ci/keep_alive bazel test --config=ci --build_tests_only --test_tag_filters=py37 doc/...
diff --git a/python/ray/util/xgboost/BUILD b/python/ray/util/xgboost/BUILD
index ecff05963..3685da3fa 100644
--- a/python/ray/util/xgboost/BUILD
+++ b/python/ray/util/xgboost/BUILD
@@ -1,26 +1,28 @@
 # --------------------------------------------------------------------
-# Tests from the python/ray/util/sgd/tests directory.
+# Tests from the python/ray/util/xgboost directory.
 # Please keep these sorted alphabetically.
 # --------------------------------------------------------------------
-# py_test(
-#     name = "simple_example",
-#     size = "small",
-#     srcs = ["simple_example.py"],
-#     deps = [":xgb_lib"],
-#     tags = ["exclusive"],
-# )
+py_test(
+ name = "simple_example",
+ size = "small",
+ srcs = ["simple_example.py"],
+ deps = [":xgb_lib"],
+ tags = ["exclusive"],
+)
+
+py_test(
+ name = "simple_tune",
+ size="small",
+ srcs = ["simple_tune.py"],
+ deps = [":xgb_lib"],
+ tags = ["exclusive"]
+)
+
+# This is a dummy test dependency that causes the above tests to be
+# re-run if any of these files changes.
+py_library(
+ name = "xgb_lib",
+ srcs = glob(["**/*.py"]),
+)
 
-# py_test(
-#     name = "simple_tune",
-#     size="small",
-#     srcs = ["simple_tune.py"],
-#     deps = [":xgb_lib"],
-#     tags = ["exlcusive"]
-# )
 
-# # This is a dummy test dependency that causes the above tests to be
-# # re-run if any of these files changes.
-# py_library(
-#     name = "xgb_lib",
-#     srcs = glob(["**/*.py"], exclude=["tests/*.py"]),
-# )
diff --git a/python/ray/util/xgboost/simple_example.py b/python/ray/util/xgboost/simple_example.py
index f48accfe6..f8881e75d 100644
--- a/python/ray/util/xgboost/simple_example.py
+++ b/python/ray/util/xgboost/simple_example.py
@@ -31,7 +31,7 @@ def main():
         train_set,
         evals=[(test_set, "eval")],
         evals_result=evals_result,
-        ray_params=RayParams(max_actor_restarts=1),
+        ray_params=RayParams(max_actor_restarts=1, num_actors=1),
         verbose_eval=False)
 
     bst.save_model("simple.xgb")
diff --git a/python/requirements/requirements_upstream.txt b/python/requirements/requirements_upstream.txt
index 6e56af5d1..5f69750d3 100644
--- a/python/requirements/requirements_upstream.txt
+++ b/python/requirements/requirements_upstream.txt
@@ -3,4 +3,4 @@
 # So we separate its own requirements file.
 
 tune-sklearn==0.2.1
-xgboost_ray==0.0.1
\ No newline at end of file
+xgboost_ray==0.0.2
\ No newline at end of file