fixing the polymorphic relationship