Fix for pip installation