Fix the benchmark