Fix for pip install bug