Store warnings under the right output

1 files changed, 4 insertions, 4 deletions

diff --git a/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithTensorFlowTestCase.java b/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithTensorFlowTestCase.java
index 5ad85ac872c..8fcd821adfd 100644
--- a/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithTensorFlowTestCase.java
+++ b/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithTensorFlowTestCase.java
@@ -91,8 +91,8 @@ public class RankingExpressionWithTensorFlowTestCase {
             fail("Expecting exception");
         }
         catch (IllegalArgumentException expected) {
-            assertEquals("Rank profile 'my_profile' is invalid: Could not import tensorflow model from tensorflow('" +
-                         modelDirectory + "','serving_defaultz'): Model does not have the specified signatures 'serving_defaultz'",
+            assertEquals("Rank profile 'my_profile' is invalid: Could not use tensorflow model from tensorflow('" +
+                         modelDirectory + "','serving_defaultz'): Model does not have the specified signature 'serving_defaultz'",
                          Exceptions.toMessageString(expected));
         }
     }
@@ -110,8 +110,8 @@ public class RankingExpressionWithTensorFlowTestCase {
             fail("Expecting exception");
         }
         catch (IllegalArgumentException expected) {
-            assertEquals("Rank profile 'my_profile' is invalid: Could not import tensorflow model from tensorflow('" +
-                         modelDirectory + "','serving_default','x'): Model does not have the specified outputs 'x'",
+            assertEquals("Rank profile 'my_profile' is invalid: Could not use tensorflow model from tensorflow('" +
+                         modelDirectory + "','serving_default','x'): Model does not have the specified output 'x'",
                          Exceptions.toMessageString(expected));
         }
     }