Fix setup.py file