Fix install stage