Revert Stage.h change