fix belongs_to