Change Stage to StageModel