Fix benchmark script.