Fix error when installing via pip