Fix stage error