diff --git a/ci/performance_tests/test_performance.py b/ci/performance_tests/test_performance.py
index f5095a716..c9d3e126c 100644
--- a/ci/performance_tests/test_performance.py
+++ b/ci/performance_tests/test_performance.py
@@ -4,7 +4,7 @@ import numpy as np
 import time
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 logger = logging.getLogger(__name__)