Fix migration belongs_to