Update setup.py

Possible fix for bug #1