diff --git a/python/ray/tune/utils/util.py b/python/ray/tune/utils/util.py
index 4d44196ce39a..d71f04fa1c86 100644
--- a/python/ray/tune/utils/util.py
+++ b/python/ray/tune/utils/util.py
@@ -475,7 +475,7 @@ def wait_for_gpu(
     .. code-block:: python
 
         def tune_func(config):
-            tune.util.wait_for_gpu()
+            tune.utils.wait_for_gpu()
             train()
 
         tuner = tune.Tuner(