Fix setup.py on Python 3