Fix python 3 compatibility in setup.py