fix batch size