Fix BatchNorm test case

1 files changed, 3 insertions, 3 deletions

diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/importer/operations/Argument.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/importer/operations/Argument.java
index ff0110a179c..8783152841d 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/importer/operations/Argument.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/integration/ml/importer/operations/Argument.java
@@ -15,10 +15,10 @@ public class Argument extends IntermediateOperation {
 
     private OrderedTensorType standardNamingType;  // using standard naming convention: d0, d1, ...
 
-    public Argument(String name, OrderedTensorType argumentType) {
+    public Argument(String name, OrderedTensorType type) {
         super(name, Collections.emptyList());
-        type = argumentType;
-        standardNamingType = OrderedTensorType.standardType(argumentType);
+        this.type = type.rename(vespaName() + "_");
+        standardNamingType = OrderedTensorType.standardType(type);
     }
 
     @Override