Fix bug
Merge pull request #778 from falinsky/patch-1

Fix bug in setup.py