1 files changed, 2 insertions, 3 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/dispatch/LoadBalancerTest.java b/container-search/src/test/java/com/yahoo/search/dispatch/LoadBalancerTest.java
index 62b56e6e8ff..4956698cc2f 100644
--- a/container-search/src/test/java/com/yahoo/search/dispatch/LoadBalancerTest.java
+++ b/container-search/src/test/java/com/yahoo/search/dispatch/LoadBalancerTest.java
@@ -7,7 +7,6 @@ import com.yahoo.search.dispatch.LoadBalancer.GroupStatus;
 import com.yahoo.search.dispatch.searchcluster.Group;
 import com.yahoo.search.dispatch.searchcluster.Node;
 import org.junit.jupiter.api.Test;
-import org.opentest4j.AssertionFailedError;
 
 import java.time.Duration;
 import java.time.Instant;
@@ -34,7 +33,7 @@ public class LoadBalancerTest {
 
         Optional<Group> grp = lb.takeGroup(null);
         Group group = grp.orElseThrow(() -> {
-            throw new AssertionFailedError("Expected a SearchCluster.Group");
+            throw new IllegalStateException("Expected a SearchCluster.Group");
         });
         assertEquals(1, group.nodes().size());
     }
@@ -47,7 +46,7 @@ public class LoadBalancerTest {
 
         Optional<Group> grp = lb.takeGroup(null);
         Group group = grp.orElseThrow(() -> {
-            throw new AssertionFailedError("Expected a SearchCluster.Group");
+            throw new IllegalStateException("Expected a SearchCluster.Group");
         });
         assertEquals(1, group.nodes().size());
     }