Record StageInfo correctly