Fixed up benchmarking script.