Adjust benchmark script to change in API.