diff --git a/.buildkite/testing.yml b/.buildkite/testing.yml
index 15bfb17..24f7c54 100644
--- a/.buildkite/testing.yml
+++ b/.buildkite/testing.yml
@@ -39,8 +39,6 @@ steps:
         agents:
           queue: "juliagpu"
           cuda: "*"
-        env:
-          RETESTITEMS_NWORKERS: 2
         if: build.message !~ /\[skip tests\]/ && build.message !~ /\[skip downstream\]/ && build.message !~ /\[skip ci\]/ && build.pull_request.labels includes "run downstream test"
         timeout_in_minutes: 60
         matrix:
@@ -161,9 +159,4 @@ steps:
               - "1"
 
 env:
-  RETESTITEMS_NWORKERS: 8
-  RETESTITEMS_NWORKER_THREADS: 2
-  RETESTITEMS_TESTITEM_TIMEOUT: 3600
-  JULIA_PKG_SERVER: ""
-  JULIA_NUM_THREADS: 4
   SECRET_CODECOV_TOKEN: "PxSr3Y7vdbiwaoX51uGykPsogxmP1IOBt5Z8TwP9GqDxIrvFocEVV2DR4Bebee12G/HYvXtQTyYXH49DpzlsfJ7ri1GQZxd9WRr+aM1DDYmzfDCfpadp4hMoJ5NQvmc/PzeGrNWOOaewaLTUP1eEaG4suygZN0lc5q9BCchIJeqoklGms5DVt/HtfTmwoD/s4wGoIJINi4RoFgnCAkzSh11hTAkyjVerfBGWEi/8E6+WBq3UKwaW4HnT02wG9qFnD4XkHpIpjMxJTpdBn5ufKI+QoJ7qJHlwqgDCtsOCblApccLTjH/BnTahNoSb/b0wdS/cblOTrtdPGzZ5UvmQ4Q==;U2FsdGVkX1/Ji2Nqeq3tqTYCBik6iXILP+rriPRqj/qxhFu4vBWWT3UnlfqDzj6oVdXyuKt0+5e+x33x2S0mBw=="
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 4f3f832..21a8b87 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -174,5 +174,3 @@ jobs:
 
 env:
   BACKEND_GROUP: "CPU"
-  RETESTITEMS_NWORKERS: 4
-  RETESTITEMS_NWORKER_THREADS: 2
diff --git a/src/internal.jl b/src/internal.jl
index 664dc52..69aa575 100644
--- a/src/internal.jl
+++ b/src/internal.jl
@@ -121,7 +121,8 @@ for op in (:get_device, :get_device_type)
 
     @eval begin
         function $(op)(x::AbstractArray{T}) where {T}
-            recursive_array_eltype(T) && return mapreduce($(op), combine_devices, x)
+            recursive_array_eltype(T) &&
+                return mapreduce(MLDataDevices.$(op), combine_devices, x)
             if hasmethod(parent, Tuple{typeof(x)})
                 parent_x = parent(x)
                 parent_x === x && return $(cpu_ret_val)
@@ -132,7 +133,7 @@ for op in (:get_device, :get_device_type)
 
         function $(op)(x::Union{Tuple, NamedTuple})
             length(x) == 0 && return $(op == :get_device ? nothing : Nothing)
-            return unrolled_mapreduce($(op), combine_devices, values(x))
+            return unrolled_mapreduce(MLDataDevices.$(op), combine_devices, values(x))
         end
     end