Add eager loading for polymorphic relations