Fix staging script