Set batch_max_size to 2000, like in MMWL