Small fix for benchmarks