Fixed an issue with setup.py