small modification on the setup.py