flax

1aff5509 - Benchmark method was previously changed, but missed few occurrences.