Update stage.hh