diff --git a/rego-java/src/main/java/com/contentgrid/opa/rego/ast/Term.java b/rego-java/src/main/java/com/contentgrid/opa/rego/ast/Term.java
index 69829df..e4b401f 100644
--- a/rego-java/src/main/java/com/contentgrid/opa/rego/ast/Term.java
+++ b/rego-java/src/main/java/com/contentgrid/opa/rego/ast/Term.java
@@ -101,7 +101,7 @@ public <T> T accept(RegoVisitor<T> visitor) {
     @EqualsAndHashCode(callSuper = true)
     public static class ArrayTerm extends Term {
 
-        List<ScalarTerm<?>> value;
+        List<Term> value;
 
         @Override
         public <T> T accept(RegoVisitor<T> visitor) {
@@ -115,7 +115,7 @@ public <T> T accept(RegoVisitor<T> visitor) {
     @EqualsAndHashCode(callSuper = true)
     public static class SetTerm extends Term {
 
-        Set<ScalarTerm<?>> value;
+        Set<Term> value;
 
         @Override
         public <T> T accept(RegoVisitor<T> visitor) {