Fix setup
Fix setup.py
LOL