diff --git a/python/ray/tune/examples/ax_example.py b/python/ray/tune/examples/ax_example.py
index 019723f7a..54081c750 100644
--- a/python/ray/tune/examples/ax_example.py
+++ b/python/ray/tune/examples/ax_example.py
@@ -113,4 +113,8 @@ if __name__ == "__main__":
     )
     algo = AxSearch(client, max_concurrent=4)
     scheduler = AsyncHyperBandScheduler(metric="hartmann6", mode="max")
-    run(easy_objective, name="ax", search_alg=algo, **config)
+    run(easy_objective,
+        name="ax",
+        search_alg=algo,
+        scheduler=scheduler,
+        **config)