Use correct batch size.