diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
index 15ea700412..f4ca274427 100644
--- a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
+++ b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
@@ -35,7 +35,7 @@ namespace raft::neighbors::cagra::detail {
 
 template <class CagraSampleFilterT>
 struct CagraSampleFilterWithQueryIdOffset {
-  const std::size_t offset;
+  const uint32_t offset;
   CagraSampleFilterT filter;
 
   CagraSampleFilterWithQueryIdOffset(const uint32_t offset, const CagraSampleFilterT filter)