Fixing pip install bug