fix stage 2