Update setup.py monkey patch to work with 2 and 3.