bench: fix up benchmarks after internal api change