1 files changed, 5 insertions, 2 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/grouping/vespa/ResultBuilderTestCase.java b/container-search/src/test/java/com/yahoo/search/grouping/vespa/ResultBuilderTestCase.java
index 6230899ec49..7e049e44f4d 100644
--- a/container-search/src/test/java/com/yahoo/search/grouping/vespa/ResultBuilderTestCase.java
+++ b/container-search/src/test/java/com/yahoo/search/grouping/vespa/ResultBuilderTestCase.java
@@ -36,7 +36,6 @@ import com.yahoo.searchlib.expression.StringResultNode;
 import org.junit.jupiter.api.Test;
 
 
-import java.util.Arrays;
 import java.util.Collections;
 import java.util.LinkedHashMap;
 import java.util.LinkedList;
@@ -916,10 +915,11 @@ public class ResultBuilderTestCase {
         assertOutput(test);
     }
 
+
     private static void assertOutput(ResultTest test) {
         RequestBuilder reqBuilder = new RequestBuilder(REQUEST_ID);
         reqBuilder.setRootOperation(GroupingOperation.fromString(test.request));
-        reqBuilder.addContinuations(Arrays.asList(test.continuation));
+        reqBuilder.addContinuations(test.getContinuations());
         reqBuilder.build();
         assertEquals(reqBuilder.getRequestList().size(), test.result.size());
 
@@ -973,6 +973,9 @@ public class ResultBuilderTestCase {
         String expectedException;
         OutputWriter outputWriter;
         Continuation continuation;
+        List<Continuation> getContinuations() {
+            return continuation != null ? List.of(continuation) : List.of();
+        }
     }
 
     private static interface OutputWriter {