Fix bug in benchmark.