fix for accidentally deleted clause that removes batch size