Fix belongs_to parameter