Fix install code