RemoveStageBarrier: minor fixes