Update setup.py

Can't install without this fix. Please check it.