have to pass in next_stage_id now, but it won't always be there so do it conditionally for now