Updated test infra.