1 files changed, 3 insertions, 3 deletions
diff --git a/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithLightGBMTestCase.java b/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithLightGBMTestCase.java
index 79d19371f1c..27fa07c7c37 100644
--- a/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithLightGBMTestCase.java
+++ b/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithLightGBMTestCase.java
@@ -48,7 +48,7 @@ public class RankingExpressionWithLightGBMTestCase {
             storedApplicationDirectory.toFile().mkdirs();
             IOUtils.copyDirectory(applicationDir.append(ApplicationPackage.MODELS_GENERATED_DIR).toFile(),
                                   storedApplicationDirectory.append(ApplicationPackage.MODELS_GENERATED_DIR).toFile());
-            RankingExpressionWithTensorFlowTestCase.StoringApplicationPackage storedApplication = new RankingExpressionWithTensorFlowTestCase.StoringApplicationPackage(storedApplicationDirectory);
+            RankingExpressionWithOnnxTestCase.StoringApplicationPackage storedApplication = new RankingExpressionWithOnnxTestCase.StoringApplicationPackage(storedApplicationDirectory);
             RankProfileSearchFixture searchFromStored = fixtureWith("lightgbm('regression.json')");
             searchFromStored.assertFirstPhaseExpression(lightGBMExpression, "my_profile");
         }
@@ -59,13 +59,13 @@ public class RankingExpressionWithLightGBMTestCase {
 
     private RankProfileSearchFixture fixtureWith(String firstPhaseExpression) {
         return fixtureWith(firstPhaseExpression, null, null,
-                new RankingExpressionWithTensorFlowTestCase.StoringApplicationPackage(applicationDir));
+                new RankingExpressionWithOnnxTestCase.StoringApplicationPackage(applicationDir));
     }
 
     private RankProfileSearchFixture fixtureWith(String firstPhaseExpression,
                                                  String constant,
                                                  String field,
-                                                 RankingExpressionWithTensorFlowTestCase.StoringApplicationPackage application) {
+                                                 RankingExpressionWithOnnxTestCase.StoringApplicationPackage application) {
         try {
             RankProfileSearchFixture fixture = new RankProfileSearchFixture(
                     application,