diff --git a/tensorflow/lite/testing/kernel_test/BUILD b/tensorflow/lite/testing/kernel_test/BUILD
index ee2403bd5c2..d1610c422b8 100644
--- a/tensorflow/lite/testing/kernel_test/BUILD
+++ b/tensorflow/lite/testing/kernel_test/BUILD
@@ -38,9 +38,6 @@ tf_cc_test(
         "//tensorflow/lite:testdata/add.bin",
         "//tensorflow/lite:testdata/test_input.csv",
     ],
-    tags = [
-        "no_oss",
-    ],
     deps = [
         ":util",
         "//tensorflow/lite/testing:tflite_driver",
diff --git a/tensorflow/lite/testing/kernel_test/util_test.cc b/tensorflow/lite/testing/kernel_test/util_test.cc
index 0599ded72d8..d9d8391bba1 100644
--- a/tensorflow/lite/testing/kernel_test/util_test.cc
+++ b/tensorflow/lite/testing/kernel_test/util_test.cc
@@ -18,7 +18,6 @@ limitations under the License.
 #include <memory>
 
 #include <gmock/gmock.h>
-#include "testing/base/public/googletest.h"
 #include <gtest/gtest.h>
 #include "tensorflow/lite/testing/tflite_driver.h"
 
@@ -32,7 +31,7 @@ TEST(UtilTest, SimpleE2ETest) {
   options.tflite_model = "tensorflow/lite/testdata/add.bin";
   options.read_input_from_file =
       "tensorflow/lite/testdata/test_input.csv";
-  options.dump_output_to_file = FLAGS_test_tmpdir + "/test_out.csv";
+  options.dump_output_to_file = ::testing::TempDir() + "/test_out.csv";
   options.kernel_type = "REFERENCE";
   std::unique_ptr<TestRunner> runner(new TfLiteDriver(
       TfLiteDriver::DelegateType::kNone, /*reference_kernel=*/true));