Updated benchmark script to reflect change in source code.