Update to MigrationCommand