Made migration tests pass