Minor changes for transform batch