diff --git a/cpp/include/raft/random/rng.cuh b/cpp/include/raft/random/rng.cuh
index 550b818c9a..ba6254bfc3 100644
--- a/cpp/include/raft/random/rng.cuh
+++ b/cpp/include/raft/random/rng.cuh
@@ -21,8 +21,8 @@
 #include "rng_state.hpp"
 #include <cassert>
 #include <optional>
+#include <raft/core/device_mdspan.hpp>
 #include <raft/core/handle.hpp>
-#include <raft/mdarray.hpp>
 #include <type_traits>
 
 namespace raft::random {