Update speculation_stage