Fixed Stage Reset bug