resolve bug StageGenerator