try fix bench.py