Small update to setup.py.