1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/tensorboard/components/tf_audio_dashboard/test/audioDashboardTests.ts b/tensorflow/tensorboard/components/tf_audio_dashboard/test/audioDashboardTests.ts
index 8cfa97e6b2..2c18a6f179 100644
--- a/tensorflow/tensorboard/components/tf_audio_dashboard/test/audioDashboardTests.ts
+++ b/tensorflow/tensorboard/components/tf_audio_dashboard/test/audioDashboardTests.ts
@@ -13,6 +13,9 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
+import * as backend_backend from '../../tf-backend/backend';
+import * as backend_router from '../../tf-backend/router';
+
 // TODO(dandelion): Fix me.
 declare function fixture(id: string): any;
 declare function stub(x, y: any): void;
@@ -22,8 +25,8 @@ describe('audio dashboard tests', () => {
   let reloadCount = 0;
   beforeEach(() => {
     audioDash = fixture('testElementFixture');
-    const router = TF.Backend.router('/data', true);
-    const backend = new TF.Backend.Backend(router);
+    const router = backend_router.router('/data', true);
+    const backend = new backend_backend.Backend(router);
     audioDash.backend = backend;
     stub('tf-audio-loader', {
       reload: () => { reloadCount++; },