diff --git a/tensorflow/lite/testing/kernel_test/BUILD b/tensorflow/lite/testing/kernel_test/BUILD
index 1a2e22c921e..0db6cfa0896 100644
--- a/tensorflow/lite/testing/kernel_test/BUILD
+++ b/tensorflow/lite/testing/kernel_test/BUILD
@@ -108,9 +108,6 @@ tf_cc_test(
     data = [
         "//tensorflow/lite:testdata/test_input.csv",
     ],
-    tags = [
-        "no_oss",
-    ],
     deps = [
         ":diff_analyzer",
         "//tensorflow/core:lib",
diff --git a/tensorflow/lite/testing/kernel_test/diff_analyzer_test.cc b/tensorflow/lite/testing/kernel_test/diff_analyzer_test.cc
index da54b38acf8..f6bb3821561 100644
--- a/tensorflow/lite/testing/kernel_test/diff_analyzer_test.cc
+++ b/tensorflow/lite/testing/kernel_test/diff_analyzer_test.cc
@@ -31,7 +31,7 @@ TEST(DiffAnalyzerTest, ZeroDiff) {
   ASSERT_EQ(diff_analyzer.ReadFiles(filename, filename), kTfLiteOk);
 
   string output_file =
-      tensorflow::io::JoinPath(FLAGS_test_tmpdir + "diff_report.csv");
+      tensorflow::io::JoinPath(::testing::TempDir(), "diff_report.csv");
   ASSERT_EQ(diff_analyzer.WriteReport(output_file), kTfLiteOk);
 
   std::string content;