RemoveStageBarrier: code clean up