Fix setup.py error