Fix setup.py in both python2 and python3