diff --git a/tests/buildkite/conftest.sh b/tests/buildkite/conftest.sh
index c159b7936..cd0744eef 100755
--- a/tests/buildkite/conftest.sh
+++ b/tests/buildkite/conftest.sh
@@ -24,8 +24,8 @@ set -x
 
 CUDA_VERSION=11.8.0
 NCCL_VERSION=2.16.5-1
-RAPIDS_VERSION=24.04
-DEV_RAPIDS_VERSION=24.06
+RAPIDS_VERSION=24.06
+DEV_RAPIDS_VERSION=24.08
 SPARK_VERSION=3.5.1
 JDK_VERSION=8
 R_VERSION=4.3.2
diff --git a/tests/cpp/helpers.cc b/tests/cpp/helpers.cc
index a47610636..fc5ec3034 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);
   }