Fixed broken setup.py

See #2