(fix) Fix benchmarks for rust API changes.