diff --git a/cpp/include/nvforest/detail/owning_buffer/gpu.hpp b/cpp/include/nvforest/detail/owning_buffer/gpu.hpp
index ba6030f..360dfde 100644
--- a/cpp/include/nvforest/detail/owning_buffer/gpu.hpp
+++ b/cpp/include/nvforest/detail/owning_buffer/gpu.hpp
@@ -10,6 +10,7 @@
 
 #include <rmm/device_buffer.hpp>
 
+#include <cuda/stream>
 #include <cuda_runtime_api.h>
 
 #include <type_traits>
@@ -26,7 +27,7 @@ struct owning_buffer<device_type::gpu, T> {
                 cudaStream_t stream) noexcept(false)
     : data_{[&device_id, &size, &stream]() {
         auto device_context = device_setter{device_id};
-        return rmm::device_buffer{size * sizeof(value_type), rmm::cuda_stream_view{stream}};
+        return rmm::device_buffer{size * sizeof(value_type), cuda::stream_ref{stream}};
       }()}
   {
   }
diff --git a/cpp/include/nvforest/handle.hpp b/cpp/include/nvforest/handle.hpp
index fd1f3bb..34e7d79 100644
--- a/cpp/include/nvforest/handle.hpp
+++ b/cpp/include/nvforest/handle.hpp
@@ -18,7 +18,7 @@ struct handle_t {
   handle_t(raft::handle_t const& raft_handle) : raft_handle_{&raft_handle} {}
   auto get_next_usable_stream() const
   {
-    return cuda_stream{raft_handle_->get_next_usable_stream().value()};
+    return cuda_stream{raft_handle_->get_next_usable_stream().get()};
   }
   auto get_stream_pool_size() const { return raft_handle_->get_stream_pool_size(); }
   auto get_usable_stream_count() const { return std::max(get_stream_pool_size(), std::size_t{1}); }