Setup.py fix
Small fix in setup.py