Fix stage4
