diff --git a/pixi.toml b/pixi.toml
index c20bdb09d6..5ba476993e 100644
--- a/pixi.toml
+++ b/pixi.toml
@@ -127,8 +127,9 @@ platforms = ["linux-64"]
 [feature.test-gpu.dependencies]
 cuda-version = "12.2.*"
 cudf = "24.04.*"
-cuspatial = "*"
 cupy = { version = "*", channel = "conda-forge" }
+cuspatial = "*"
+rmm = { version = "*", channel = "rapidsai" }
 ucx = { version = "*", channel = "conda-forge" }
 
 [feature.test-gpu.tasks]