Fix the benchmark tool