Added override for nextstage