Fix the run benchmark script