1 files changed, 2 insertions, 2 deletions
diff --git a/searchlib/src/vespa/searchlib/features/tensor_from_attribute_executor.h b/searchlib/src/vespa/searchlib/features/tensor_from_attribute_executor.h
index 5a3fede76e8..7b04d10cea2 100644
--- a/searchlib/src/vespa/searchlib/features/tensor_from_attribute_executor.h
+++ b/searchlib/src/vespa/searchlib/features/tensor_from_attribute_executor.h
@@ -28,9 +28,9 @@ private:
 
 public:
     TensorFromAttributeExecutor(const search::attribute::IAttributeVector *attribute,
-                                const vespalib::string &dimension)
+                                const vespalib::eval::ValueType &valueType)
         : _attribute(attribute),
-          _type(vespalib::eval::ValueType::make_type(CellType::DOUBLE, {{dimension}})),
+          _type(valueType),
           _attrBuffer(),
           _addr_ref(),
           _tensor()