Fixed setup.py issue