bugfix in benchmark.py