This doesn't look right. Maybe missed during the rebase? `batch_size` isn't part of the `#base_state{}` record any more.