Fix Batch scenario bug