Fixed bug with pip install