Fix syntax issue on Python 3