Fix bench.py