Small edit to StageI.java