Fix install function