Fix foreign key in migration