Code required to enable unit tests on infra