Setup.py fix