Update support for assign_attributes.