fix bug in setup.py