Added migration tests