Update migration tests