Fix benchmarks for new API