Small update to stages