Fix issue when calling bench.py from @r-gaia-cs