Fix benchmarking