Minor fix in transformer