Fix minor benchmarking issue