- benchmark.py fix