Fix up setup.py