1 files changed, 3 insertions, 4 deletions
diff --git a/streamingvisitors/src/vespa/searchvisitor/hitcollector.cpp b/streamingvisitors/src/vespa/searchvisitor/hitcollector.cpp
index f3850154505..12dce7d376c 100644
--- a/streamingvisitors/src/vespa/searchvisitor/hitcollector.cpp
+++ b/streamingvisitors/src/vespa/searchvisitor/hitcollector.cpp
@@ -4,8 +4,7 @@
 #include <vespa/searchlib/fef/feature_resolver.h>
 #include <vespa/vespalib/util/stringfmt.h>
 #include <algorithm>
-#include <vespa/eval/eval/tensor.h>
-#include <vespa/eval/eval/tensor_engine.h>
+#include <vespa/eval/eval/engine_or_factory.h>
 #include <vespa/vespalib/objects/nbostream.h>
 
 #include <vespa/log/log.h>
@@ -163,9 +162,9 @@ HitCollector::getFeatureSet(IRankProgram &rankProgram,
         for (uint32_t j = 0; j < names.size(); ++j) {
             if (resolver.is_object(j)) {
                 auto obj = resolver.resolve(j).as_object(docId);
-                if (const auto *tensor = obj.get().as_tensor()) {
+                if (! obj.get().is_double()) {
                     vespalib::nbostream buf;
-                    tensor->engine().encode(*tensor, buf);
+                    vespalib::eval::EngineOrFactory::get().encode(obj.get(), buf);
                     f[j].set_data(vespalib::Memory(buf.peek(), buf.size()));
                 } else {
                     f[j].set_double(obj.get().as_double());