Minor improvement for the benchmark