Update migration, too