diff --git a/python/ray/tests/test_dask_scheduler.py b/python/ray/tests/test_dask_scheduler.py
index 28a98a76e..54ba40521 100644
--- a/python/ray/tests/test_dask_scheduler.py
+++ b/python/ray/tests/test_dask_scheduler.py
@@ -35,7 +35,9 @@ def test_ray_dask_basic(ray_start_regular_shared):
 def test_ray_dask_persist(ray_start_regular_shared):
     arr = da.ones(5) + 2
     result = arr.persist(scheduler=ray_dask_get)
-    np.testing.assert_array_equal(result.dask.values()[0], np.ones(5) + 2)
+    np.testing.assert_array_equal(
+        next(iter(result.dask.values())),
+        np.ones(5) + 2)
 
 
 if __name__ == "__main__":