Fix setup from pip install