Maintain TensorType in documents

1 files changed, 3 insertions, 2 deletions

diff --git a/document/src/test/java/com/yahoo/vespaxmlparser/VespaXmlFieldReaderTestCase.java b/document/src/test/java/com/yahoo/vespaxmlparser/VespaXmlFieldReaderTestCase.java
index 3cfbcac5b62..3dc6ebd1403 100644
--- a/document/src/test/java/com/yahoo/vespaxmlparser/VespaXmlFieldReaderTestCase.java
+++ b/document/src/test/java/com/yahoo/vespaxmlparser/VespaXmlFieldReaderTestCase.java
@@ -12,6 +12,7 @@ import com.yahoo.document.predicate.FeatureRange;
 import com.yahoo.document.predicate.FeatureSet;
 import com.yahoo.document.predicate.Predicate;
 import com.yahoo.document.serialization.DeserializationException;
+import com.yahoo.tensor.TensorType;
 import org.apache.commons.codec.binary.Base64;
 import org.junit.Test;
 
@@ -75,8 +76,8 @@ public class VespaXmlFieldReaderTestCase {
 
     @Test
     public void requireThatPutsForTensorFieldsAreNotSupported() throws Exception {
-        assertThrows(new Field("my_tensor", DataType.TENSOR), "",
-                "Field 'my_tensor': XML input for fields of type TENSOR is not supported. Please use JSON input instead.");
+        assertThrows(new Field("my_tensor", new TensorDataType(TensorType.empty)), "",
+                     "Field 'my_tensor': XML input for fields of type TENSOR is not supported. Please use JSON input instead.");
     }
 
     private class MockedReaderFixture {