Made the new change for PrimaryStage