minor adjustments to current stage