diff --git a/runtime/src/main/java/dev/cel/runtime/CelUnknownSet.java b/runtime/src/main/java/dev/cel/runtime/CelUnknownSet.java index 070e94cf0..05201d1e5 100644 --- a/runtime/src/main/java/dev/cel/runtime/CelUnknownSet.java +++ b/runtime/src/main/java/dev/cel/runtime/CelUnknownSet.java @@ -16,6 +16,7 @@ import com.google.auto.value.AutoValue; import com.google.common.collect.ImmutableSet; +import com.google.common.collect.Sets; /** * Unknown set representation. @@ -65,14 +66,8 @@ private static CelUnknownSet create( public static CelUnknownSet union(CelUnknownSet lhs, CelUnknownSet rhs) { return create( - ImmutableSet.builder() - .addAll(lhs.attributes()) - .addAll(rhs.attributes()) - .build(), - ImmutableSet.builder() - .addAll(lhs.unknownExprIds()) - .addAll(rhs.unknownExprIds()) - .build()); + Sets.union(lhs.attributes(), rhs.attributes()).immutableCopy(), + Sets.union(lhs.unknownExprIds(), rhs.unknownExprIds()).immutableCopy()); } public CelUnknownSet merge(CelUnknownSet rhs) {