Added support for polymorphic belongs_to association