Remove duplicate code (set_belongs_to_association_for)