Update the benchmark to work with the latest master.