Fix the setup.py definition