diff --git a/benchmarks/device_uvector/device_uvector_bench.cu b/benchmarks/device_uvector/device_uvector_bench.cu
index 701e5fceb..454db81a5 100644
--- a/benchmarks/device_uvector/device_uvector_bench.cu
+++ b/benchmarks/device_uvector/device_uvector_bench.cu
@@ -15,7 +15,6 @@
  */
 
 #include "../synchronization/synchronization.hpp"
-#include "thrust/detail/raw_pointer_cast.h"
 
 #include <rmm/cuda_stream.hpp>
 #include <rmm/detail/error.hpp>
@@ -30,6 +29,9 @@
 
 #include <cuda_runtime_api.h>
 
+#include <thrust/device_vector.h>
+#include <thrust/memory.h>
+
 #include <cstdio>
 #include <type_traits>
 
diff --git a/include/rmm/mr/device/thrust_allocator_adaptor.hpp b/include/rmm/mr/device/thrust_allocator_adaptor.hpp
index 56e910801..c09c9f334 100644
--- a/include/rmm/mr/device/thrust_allocator_adaptor.hpp
+++ b/include/rmm/mr/device/thrust_allocator_adaptor.hpp
@@ -19,8 +19,9 @@
 #include <rmm/mr/device/device_memory_resource.hpp>
 #include <rmm/mr/device/per_device_resource.hpp>
 
-#include <thrust/detail/type_traits/pointer_traits.h>
 #include <thrust/device_malloc_allocator.h>
+#include <thrust/device_ptr.h>
+#include <thrust/memory.h>
 
 namespace rmm::mr {
 /**
diff --git a/tests/mr/device/thrust_allocator_tests.cu b/tests/mr/device/thrust_allocator_tests.cu
index e0d467595..3a56a541b 100644
--- a/tests/mr/device/thrust_allocator_tests.cu
+++ b/tests/mr/device/thrust_allocator_tests.cu
@@ -23,6 +23,8 @@
 
 #include <gtest/gtest.h>
 
+#include <thrust/reduce.h>
+
 // explicit instantiation for test coverage purposes
 template class rmm::mr::thrust_allocator<int>;