diff --git a/tests/python/test_data_iterator.py b/tests/python/test_data_iterator.py
index 1e43fcf774ce..52b2274ddb4e 100644
--- a/tests/python/test_data_iterator.py
+++ b/tests/python/test_data_iterator.py
@@ -111,7 +111,7 @@ def run_data_iterator(
     )
     arr_predt = from_arrays.predict(Xy)
 
-    if tree_method != "gpu_hist":
+    if tree_method == "hist":
         rtol = 1e-1  # flaky
     else:
         # Model can be sensitive to quantiles, use 1e-2 to relax the test.