Fix for error in setup.py