Fixed issue with setup.py