Update setup.py

Reverting last commit.