Small fix in setup.py