diff --git a/tests/buildkite/conftest.sh b/tests/buildkite/conftest.sh
index 41b1ea0e879c..a0462d738754 100755
--- a/tests/buildkite/conftest.sh
+++ b/tests/buildkite/conftest.sh
@@ -24,7 +24,7 @@ set -x
 
 CUDA_VERSION=11.8.0
 NCCL_VERSION=2.16.5-1
-RAPIDS_VERSION=24.04
+RAPIDS_VERSION=24.06
 DEV_RAPIDS_VERSION=24.06
 SPARK_VERSION=3.5.1
 JDK_VERSION=8
diff --git a/tests/cpp/helpers.cc b/tests/cpp/helpers.cc
index a4761063688d..fc5ec303448d 100644
--- a/tests/cpp/helpers.cc
+++ b/tests/cpp/helpers.cc
@@ -728,7 +728,7 @@ class RMMAllocator {
     for (int i = 0; i < n_gpu; ++i) {
       CHECK_EQ(cudaSetDevice(i), cudaSuccess);
       cuda_mr.push_back(std::make_unique<CUDAMemoryResource>());
-      pool_mr.push_back(std::make_unique<PoolMemoryResource>(cuda_mr[i].get()));
+      pool_mr.push_back(std::make_unique<PoolMemoryResource>(cuda_mr[i].get(), 0ul));
     }
     CHECK_EQ(cudaSetDevice(current_device), cudaSuccess);
   }