FIX: Fix setup.py