diff --git a/tensorflow/lite/testing/kernel_test/BUILD b/tensorflow/lite/testing/kernel_test/BUILD
index d1610c422b8..1a2e22c921e 100644
--- a/tensorflow/lite/testing/kernel_test/BUILD
+++ b/tensorflow/lite/testing/kernel_test/BUILD
@@ -84,9 +84,6 @@ cc_test(
         "//tensorflow/lite:testdata/multi_add.bin",
         "//tensorflow/lite:testdata/test_input.csv",
     ],
-    tags = [
-        "no_oss",
-    ],
     deps = [
         ":input_generator",
         "@com_google_googletest//:gtest_main",
diff --git a/tensorflow/lite/testing/kernel_test/input_generator_test.cc b/tensorflow/lite/testing/kernel_test/input_generator_test.cc
index 2223a9196ff..1719aa7bc96 100644
--- a/tensorflow/lite/testing/kernel_test/input_generator_test.cc
+++ b/tensorflow/lite/testing/kernel_test/input_generator_test.cc
@@ -18,7 +18,6 @@ limitations under the License.
 #include <map>
 
 #include <gmock/gmock.h>
-#include "testing/base/public/googletest.h"
 #include <gtest/gtest.h>
 
 namespace tflite {
@@ -47,7 +46,7 @@ TEST(InputGeneratorTest, ReadWriteSimpleFile) {
   inputs.push_back(content);
   ASSERT_EQ(input_generator.GetInputs(), inputs);
 
-  auto output_filename = FLAGS_test_tmpdir + "/out.csv";
+  auto output_filename = ::testing::TempDir() + "/out.csv";
   ASSERT_EQ(input_generator.WriteInputsToFile(output_filename), kTfLiteOk);
 
   std::ifstream in(output_filename);