Python 3 fix in setup.py