Setup.py correction