Update setup.py

Same as earlier commit