1 files changed, 1 insertions, 3 deletions
diff --git a/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp b/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp
index 7a200a46ab2..826984832f6 100644
--- a/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp
+++ b/searchlib/src/tests/features/onnx_feature/onnx_feature_test.cpp
@@ -58,9 +58,7 @@ struct OnnxFeatureTest : ::testing::Test {
         indexEnv.getProperties().add(expr_name, expr);
     }
     void add_onnx(const vespalib::string &name, const vespalib::string &file) {
-        vespalib::string feature_name = onnx_feature(name);
-        vespalib::string file_name = feature_name + ".fileref";
-        indexEnv.getProperties().add(file_name, file);
+        indexEnv.addOnnxModel(name, file);
     }
     void compile(const vespalib::string &seed) {
         resolver->addSeed(seed);