1 files changed, 5 insertions, 5 deletions
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/zone/ZoneFilterMock.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/zone/ZoneFilterMock.java
index e68bf0ccc24..a7d51fa4d24 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/zone/ZoneFilterMock.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/zone/ZoneFilterMock.java
@@ -46,17 +46,17 @@ public class ZoneFilterMock implements ZoneList {
     }
 
     @Override
-    public ZoneList in(Environment environment) {
-        return filter(zoneId -> zoneId.environment() == environment);
+    public ZoneList in(Environment... environments) {
+        return filter(zoneId -> new HashSet<>(Arrays.asList(environments)).contains(zoneId.environment()));
     }
 
     @Override
-    public ZoneList in(RegionName region) {
-        return filter(zoneId -> zoneId.region().equals(region));
+    public ZoneList in(RegionName... regions) {
+        return filter(zoneId -> new HashSet<>(Arrays.asList(regions)).contains(zoneId.region()));
     }
 
     @Override
-    public ZoneList zones(ZoneId... zones) {
+    public ZoneList among(ZoneId... zones) {
         return filter(zoneId -> new HashSet<>(Arrays.asList(zones)).contains(zoneId));
     }