Fixed bug introduced in transformer class.