[WIP] Recover migration tests