RemoveStageBarrier: bug fix