Fix syntax for Python 3