BUG: fixed the bug in the setup.py