minor fix to benchmarking code