Fix breakage in setup.py