Update StageInfo.scala