diff --git a/benchmarks/bench/run_length_encode/encode.cu b/benchmarks/bench/run_length_encode/encode.cu
index 8d22311d1..2dc94309b 100644
--- a/benchmarks/bench/run_length_encode/encode.cu
+++ b/benchmarks/bench/run_length_encode/encode.cu
@@ -91,7 +91,7 @@ static void rle(nvbench::state &state, nvbench::type_list<T, OffsetT>)
                                               accum_t,
                                               device_reduce_by_key_policy_hub>;
   #else
-  using policy_t = cub::detail::device_run_length_encode_policy_hub<accum_t, key_t>;
+  using policy_t = cub::detail::device_run_length_encode_policy_hub<accum_t, T>;
   using dispatch_t = cub::DispatchReduceByKey<keys_input_it_t,
                                               unique_output_it_t,
                                               vals_input_it_t,