Fix a Python 3 issue