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