Fixed benchmark

Benchmark was broken because of old implementation