fix dialog doesn't disappear in stage 2-3