Fix error in setup.py
Support custom prefix