Fix stage naming