1 files changed, 6 insertions, 2 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/evaluation/VariableTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/evaluation/VariableTensor.java
index 1f6ad050368..6c149724aca 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/evaluation/VariableTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/evaluation/VariableTensor.java
@@ -3,6 +3,7 @@ package com.yahoo.tensor.evaluation;
 
 import com.google.common.annotations.Beta;
 import com.yahoo.tensor.Tensor;
+import com.yahoo.tensor.TensorType;
 import com.yahoo.tensor.functions.PrimitiveTensorFunction;
 import com.yahoo.tensor.functions.TensorFunction;
 import com.yahoo.tensor.functions.ToStringContext;
@@ -25,15 +26,18 @@ public class VariableTensor extends PrimitiveTensorFunction {
     }
 
     @Override
-    public List<TensorFunction> functionArguments() { return Collections.emptyList(); }
+    public List<TensorFunction> arguments() { return Collections.emptyList(); }
 
     @Override
-    public TensorFunction replaceArguments(List<TensorFunction> arguments) { return this; }
+    public TensorFunction withArguments(List<TensorFunction> arguments) { return this; }
 
     @Override
     public PrimitiveTensorFunction toPrimitive() { return this; }
 
     @Override
+    public TensorType type(EvaluationContext context) { return context.getTensorType(name); }
+
+    @Override
     public Tensor evaluate(EvaluationContext context) {
         return context.getTensor(name);
     }