fixed issues with bench.py