Fix for setup.py