1 files changed, 2 insertions, 4 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java b/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java
index fdbde08d926..19fbc116558 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/processing/OnnxModelConfigGenerator.java
@@ -86,10 +86,8 @@ public class OnnxModelConfigGenerator extends Processor {
                     }
 
                     OnnxModel onnxModel = schema.onnxModels().get(modelConfigName);
-                    if (onnxModel == null) {
-                        onnxModel = new OnnxModel(modelConfigName, path);
-                        schema.onnxModels().add(onnxModel);
-                    }
+                    if (onnxModel == null)
+                        schema.add(new OnnxModel(modelConfigName, path));
                 }
             }
         }