Fix install bug