diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
index 13efa277ed..15ea700412 100644
--- a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
+++ b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
@@ -38,7 +38,7 @@ struct CagraSampleFilterWithQueryIdOffset {
   const std::size_t offset;
   CagraSampleFilterT filter;
 
-  CagraSampleFilterWithQueryIdOffset(const std::size_t offset, const CagraSampleFilterT filter)
+  CagraSampleFilterWithQueryIdOffset(const uint32_t offset, const CagraSampleFilterT filter)
     : offset(offset), filter(filter)
   {
   }