Fix mismatched benchmark code