Fixed errors in the migration