diff --git a/tensorflow/compiler/xla/BUILD b/tensorflow/compiler/xla/BUILD
index dd9f83bf26e..01f35df0e20 100644
--- a/tensorflow/compiler/xla/BUILD
+++ b/tensorflow/compiler/xla/BUILD
@@ -722,6 +722,7 @@ tf_cc_test(
         ":text_literal_writer",
         ":types",
         "//tensorflow/core:lib",
+        "//tensorflow/core:test",
         "//tensorflow/core:test_main",
     ],
 )
diff --git a/tensorflow/compiler/xla/text_literal_writer_test.cc b/tensorflow/compiler/xla/text_literal_writer_test.cc
index 5cbaf2fcc19..667d6296117 100644
--- a/tensorflow/compiler/xla/text_literal_writer_test.cc
+++ b/tensorflow/compiler/xla/text_literal_writer_test.cc
@@ -23,6 +23,7 @@ limitations under the License.
 #include "tensorflow/compiler/xla/test.h"
 #include "tensorflow/compiler/xla/test_helpers.h"
 #include "tensorflow/compiler/xla/types.h"
+#include "tensorflow/core/lib/core/status_test_util.h"
 #include "tensorflow/core/lib/io/path.h"
 #include "tensorflow/core/platform/env.h"
 #include "tensorflow/core/platform/logging.h"
@@ -35,12 +36,12 @@ TEST(TextLiteralWriterTest, WritesFloatLiteral) {
       {3.14, 2.17},
       {1.23, 4.56},
   });
-  string path =
-      tensorflow::io::JoinPath(tensorflow::testing::TmpDir(), "/whatever");
+  string path;
+  ASSERT_TRUE(tensorflow::Env::Default()->LocalTempFilename(&path));
   ASSERT_IS_OK(TextLiteralWriter::WriteToPath(literal, path));
   string contents;
-  TF_CHECK_OK(tensorflow::ReadFileToString(tensorflow::Env::Default(), path,
-                                           &contents));
+  TF_ASSERT_OK(tensorflow::ReadFileToString(tensorflow::Env::Default(), path,
+                                            &contents));
   const string expected = R"(f32[2,2]
 (0, 0): 3.14
 (0, 1): 2.17