Fix for installing via pip