Fixed LoadStage code