Small fix to belongs_to relationships.