diff --git a/testing/src/main/java/dev/cel/testing/testrunner/CelTestSuite.java b/testing/src/main/java/dev/cel/testing/testrunner/CelTestSuite.java
index 0198574a4..e6086f128 100644
--- a/testing/src/main/java/dev/cel/testing/testrunner/CelTestSuite.java
+++ b/testing/src/main/java/dev/cel/testing/testrunner/CelTestSuite.java
@@ -47,6 +47,8 @@ public abstract static class Builder {
 
     public abstract Builder setSections(Set<CelTestSection> section);
 
+    public abstract Builder setSections(CelTestSection... sections);
+
     public abstract Builder setSource(Source source);
 
     @CheckReturnValue
@@ -80,6 +82,8 @@ public abstract static class Builder {
 
       public abstract Builder setTests(Set<CelTestCase> tests);
 
+      public abstract Builder setTests(CelTestCase... tests);
+
       public abstract Builder setDescription(String description);
 
       @CheckReturnValue
diff --git a/testing/src/test/java/dev/cel/testing/testrunner/BUILD.bazel b/testing/src/test/java/dev/cel/testing/testrunner/BUILD.bazel
index bee44b4dd..db1202aa8 100644
--- a/testing/src/test/java/dev/cel/testing/testrunner/BUILD.bazel
+++ b/testing/src/test/java/dev/cel/testing/testrunner/BUILD.bazel
@@ -11,9 +11,9 @@ package(
     ],
 )
 
-# Since the user test class is triggered by the cel_test_runner rule, we should not add it to the
+# Since the user test class is triggered by the cel_test rule, we should not add it to the
 # junit4_test_suite.
-# This is just a sample test class for the cel_test_runner rule.
+# This is just a sample test class for the cel_test rule.
 java_library(
     name = "user_test",
     srcs = ["UserTest.java"],
@@ -26,7 +26,7 @@ java_library(
     ],
 )
 
-# This is just a sample test class for the cel_test_runner rule.
+# This is just a sample test class for the cel_test rule.
 java_library(
     name = "env_config_user_test",
     srcs = ["EnvConfigUserTest.java"],