Another fix for pip install