Small fix to benchmark script