1 files changed, 6 insertions, 6 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java
index 89e981df49e..54e83fa472f 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java
@@ -117,9 +117,9 @@ public class Generate<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAM
     }
 
     @Override
-    public String toString(ToStringContext context) { return type + "(" + generatorToString(context) + ")"; }
+    public String toString(ToStringContext<NAMETYPE> context) { return type + "(" + generatorToString(context) + ")"; }
 
-    private String generatorToString(ToStringContext context) {
+    private String generatorToString(ToStringContext<NAMETYPE> context) {
         if (freeGenerator != null)
             return freeGenerator.toString();
         else
@@ -183,11 +183,11 @@ public class Generate<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAM
     }
 
     /** A context which adds the bindings of the generate dimension names to the given context. */
-    private class GenerateToStringContext implements ToStringContext {
+    private class GenerateToStringContext implements ToStringContext<NAMETYPE> {
 
-        private final ToStringContext context;
+        private final ToStringContext<NAMETYPE> context;
 
-        public GenerateToStringContext(ToStringContext context) {
+        public GenerateToStringContext(ToStringContext<NAMETYPE> context) {
             this.context = context;
         }
 
@@ -200,7 +200,7 @@ public class Generate<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAM
         }
 
         @Override
-        public ToStringContext parent() { return context; }
+        public ToStringContext<NAMETYPE> parent() { return context; }
 
     }