Add support for changing the class of an polymorphic embedded document when setting values