Fixing polymorphic relationship