diff --git a/tensorboard/plugins/mesh/BUILD b/tensorboard/plugins/mesh/BUILD
index 8472cd2422..b1c6422aec 100644
--- a/tensorboard/plugins/mesh/BUILD
+++ b/tensorboard/plugins/mesh/BUILD
@@ -145,6 +145,7 @@ py_test(
     srcs = ["demo_utils_test.py"],
     data = [
         ":test_data",
+        "//tensorboard/compat:tensorflow",
     ],
     srcs_version = "PY2AND3",
     deps = [
diff --git a/tensorboard/plugins/mesh/demo_utils_test.py b/tensorboard/plugins/mesh/demo_utils_test.py
index d06befe629..47ad137dd6 100644
--- a/tensorboard/plugins/mesh/demo_utils_test.py
+++ b/tensorboard/plugins/mesh/demo_utils_test.py
@@ -20,10 +20,13 @@
 
 import os
 import unittest
+
+import tensorflow as tf
+
 from tensorboard.plugins.mesh import demo_utils
 
 
-class TestPLYReader(unittest.TestCase):
+class TestPLYReader(tf.test.TestCase):
   def test_parse_vertex(self):
     """Tests vertex coordinate and color parsing."""
     # Vertex 3D coordinates with RGBA color.
@@ -56,4 +59,4 @@ def test_read_ascii_ply(self):
 
 
 if __name__ == '__main__':
-  unittest.main()
+  tf.test.main()