Add more specs for polymorphic associations