RemoveStageBarrier: fix shuffle map stage fail over