fix bug in benchmark script