Update setup.py

Reverting back to the original repo.