Fix optional transform function