Fix issue with setup.py