Fix the benchmark.

Broke it during the "cleanup"...