Minor update on `setup.py`