Update setup.py

Missed out on one change.
Merge pull request #4 from lowks/patch-2

Update setup.py