Fix benchmarks

Probably good to just merge straight in.