Tweak benchmark to be usable as a test