Fix resume stage bug!