Minor code cleanup for AtomicBatchWrites.