Small fixup for setup.py