[fix] Fix setup.py