1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/tensorboard/components/tf_audio_dashboard/demo/index.html b/tensorflow/tensorboard/components/tf_audio_dashboard/demo/index.html
index 177bc85db0..dc8cd91d43 100644
--- a/tensorflow/tensorboard/components/tf_audio_dashboard/demo/index.html
+++ b/tensorflow/tensorboard/components/tf_audio_dashboard/demo/index.html
@@ -42,14 +42,16 @@ limitations under the License.
         <tf-audio-dashboard id="demo" backend="[[backend]]"></tf-audio-dashboard>
       </template>
       <script>
+        import {Backend} from '../tf-backend/backend';
+        import {router} from '../tf-backend/router';
+
         Polymer({
           is: "audio-dash-demo",
           properties: {
             backend: {
               type: Object,
               value: function() {
-                var router = new TF.Backend.router("/data", true);
-                return new TF.Backend.Backend(router);
+                return new Backend(new router("/data", true));
               },
             },
           },