Fix error in the migration