fix migration tests