Fix the benchmark runner