Updated setup.py
Removed bug which leads to infinite loop