Update setup.py for fork