Small change to the migration