Fixed belongs_to field