修改 batch 判斷