Fixed slight problem with setup.py