Added support for polymorphic models in collections