Test correct output in batch norm case

1 files changed, 1 insertions, 3 deletions

diff --git a/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java b/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java
index 6eb13ebbd6c..bf9684082f4 100644
--- a/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java
+++ b/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java
@@ -23,9 +23,7 @@ public class BatchNormImportTestCase {
         RankingExpression output = signature.outputExpression("y");
         assertNotNull(output);
         assertEquals("dnn/batch_normalization_3/batchnorm/add_1", output.getName());
-//        model.assertEqualResult("X", output.getName());
-
-        model.assertEqualResult("X", "dnn/outputs/BiasAdd");
+        model.assertEqualResult("X", output.getName());
     }
 
 }