Fix benchmark script