RemoveStageBarrier: fix bug