small changes while working on benchmarks