Fix up install code