Migration update

Small update to a migration.