diff --git a/kernels/test/op_native_batch_norm_test.cpp b/kernels/test/op_native_batch_norm_test.cpp
index 34c3ef1a36..8c9581c66d 100644
--- a/kernels/test/op_native_batch_norm_test.cpp
+++ b/kernels/test/op_native_batch_norm_test.cpp
@@ -140,7 +140,7 @@ class OpNativeBatchNormLegitNoTrainingOutTest : public OperatorTest {
           2e-2,
           executorch::runtime::testing::internal::kDefaultAtol);
       EXPECT_TENSOR_CLOSE_WITH_TOL(
-          out1,
+          out2,
           out2_expected,
           2e-2,
           executorch::runtime::testing::internal::kDefaultAtol);