Fix minor issue with the benchmark code