Update WormStage.java