fixed stageButton bug when coming back from a stage