Fix python 3 syntax in benchmark.py