RemoveStageBarrier: fix test code