Correction on setup.py