Bug fix to benchmarking script