Fix setup.py issue