Small fix in setup
Fixing setup.py