diff --git a/tensorflow/lite/micro/benchmarks/BUILD b/tensorflow/lite/micro/benchmarks/BUILD
index 888c4da96ba..4394a9b4dc6 100644
--- a/tensorflow/lite/micro/benchmarks/BUILD
+++ b/tensorflow/lite/micro/benchmarks/BUILD
@@ -15,6 +15,9 @@ cc_library(
     hdrs = [
         "micro_benchmark.h",
     ],
+    visibility = [
+        "//visibility:public",
+    ],
     deps = [
         "//tensorflow/lite/micro:micro_error_reporter",
         "//tensorflow/lite/micro:micro_framework",
diff --git a/tensorflow/lite/micro/benchmarks/micro_benchmark.h b/tensorflow/lite/micro/benchmarks/micro_benchmark.h
index 2eb3787e01d..272c7207739 100644
--- a/tensorflow/lite/micro/benchmarks/micro_benchmark.h
+++ b/tensorflow/lite/micro/benchmarks/micro_benchmark.h
@@ -43,7 +43,7 @@ class MicroBenchmarkRunner {
   void RunSingleIteration() {
     // Run the model on this input and make sure it succeeds.
     TfLiteStatus invoke_status = interpreter_.Invoke();
-    if (invoke_status != kTfLiteOk) {
+    if (invoke_status == kTfLiteError) {
       MicroPrintf("Invoke failed.");
     }
   }