Sync benchmarks to latest library state.

I started working on benchmarking the transformer instances and the benchmarks were broken by the latest updates, here is the fix.