Merge "Update init_first_stage"