Fixing part of the setup.py file