1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/tensorboard/components/vz_heatmap/demo/index.html b/tensorflow/tensorboard/components/vz_heatmap/demo/index.html
index 63dd04485f..0ef092fccc 100644
--- a/tensorflow/tensorboard/components/vz_heatmap/demo/index.html
+++ b/tensorflow/tensorboard/components/vz_heatmap/demo/index.html
@@ -22,6 +22,7 @@ limitations under the License.
   <title>Heatmap example</title>
   <link rel="import" href="../vz-heatmap.html">
   <link rel="import" href="../../iron-demo-helpers/demo-snippet.html">
+  <link rel="import" href="../../tf-imports/d3.html">
 </head>
 <script>
   function generateRandomMatrix() {
@@ -108,7 +109,7 @@ limitations under the License.
     setTimeout(function () {
       var heatmapColor = document.getElementById('color_function');
       heatmapColor.colorFunction =
-          d3.scale.linear().range(['white', 'black']).domain([0, 5]);
+          d3.scaleLinear().range(['white', 'black']).domain([0, 5]);
     }, 1500);
   </script>
 </div>