1 files changed, 2 insertions, 5 deletions
diff --git a/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp b/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp
index d4353532309..636c949be08 100644
--- a/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp
+++ b/searchlib/src/vespa/searchlib/tensor/dense_tensor_attribute.cpp
@@ -112,7 +112,7 @@ void
 DenseTensorAttribute::internal_set_tensor(DocId docid, const vespalib::eval::Value& tensor)
 {
     consider_remove_from_index(docid);
-    EntryRef ref = _denseTensorStore.setTensor(tensor);
+    EntryRef ref = _denseTensorStore.store_tensor(tensor);
     setTensorRef(docid, ref);
 }
 
@@ -229,10 +229,7 @@ DenseTensorAttribute::getTensor(DocId docId) const
     if (docId < getCommittedDocIdLimit()) {
         ref = acquire_entry_ref(docId);
     }
-    if (!ref.valid()) {
-        return {};
-    }
-    return _denseTensorStore.getTensor(ref);
+    return _denseTensorStore.get_tensor(ref);
 }
 
 vespalib::eval::TypedCells