1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/tensorboard/components/tf_graph_controls/demo/BUILD b/tensorflow/tensorboard/components/tf_graph_controls/demo/BUILD
index c47cb90a03..cd86ac7320 100644
--- a/tensorflow/tensorboard/components/tf_graph_controls/demo/BUILD
+++ b/tensorflow/tensorboard/components/tf_graph_controls/demo/BUILD
@@ -1,19 +1,19 @@
 package(default_visibility = ["//tensorflow:internal"])
 
-load("@io_bazel_rules_closure//closure:defs.bzl", "web_library")
+load("//tensorflow/tensorboard:web.bzl", "ts_web_library")
 
 licenses(["notice"])  # Apache 2.0
 
 # bazel run //third_party/tensorflow/tensorboard/components/tf_graph_controls/demo
-web_library(
+ts_web_library(
     name = "demo",
     srcs = ["index.html"],
     path = "/tf-graph-controls/demo",
     deps = [
         "//tensorflow/tensorboard/components/tf_graph_controls",
+        "//tensorflow/tensorboard/components/tf_imports:webcomponentsjs",
         "@org_polymer_iron_demo_helpers",
         "@org_polymer_paper_styles",
-        "@org_polymer_webcomponentsjs",
     ],
 )