Corrected an issue with the benchmark script