Handle polymorphic associations
TODO comment for polymorphic associations test