minor fix for correctness benchmarks