fix batch size problem