fix up the benchmark script

[skip ci]