Unit test infra update