Remove ZoneRegistryMock from services.xml in tests

1 files changed, 5 insertions, 6 deletions

diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/restapi/configserver/ConfigServerApiHandlerTest.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/restapi/configserver/ConfigServerApiHandlerTest.java
index e723b5a2f30..8aea26f21e3 100644
--- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/restapi/configserver/ConfigServerApiHandlerTest.java
+++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/restapi/configserver/ConfigServerApiHandlerTest.java
@@ -38,12 +38,11 @@ public class ConfigServerApiHandlerTest extends ControllerContainerTest {
 
     @Before
     public void before() {
-        ZoneRegistryMock zoneRegistry = (ZoneRegistryMock) container.components()
-                .getComponent(ZoneRegistryMock.class.getName());
-        zoneRegistry.setDefaultRegionForEnvironment(Environment.dev, RegionName.from("us-north-2"))
-                .setZones(zones);
-        this.tester = new ContainerTester(container, responseFiles);
-        this.proxy = (ConfigServerProxyMock) container.components().getComponent(ConfigServerProxyMock.class.getName());
+        tester = new ContainerTester(container, responseFiles);
+        tester.serviceRegistry().zoneRegistry()
+              .setDefaultRegionForEnvironment(Environment.dev, RegionName.from("us-north-2"))
+              .setZones(zones);
+        proxy = (ConfigServerProxyMock) container.components().getComponent(ConfigServerProxyMock.class.getName());
     }
 
     @Test