         .range([0, @height * .90])
 
       # update the node label
-      @nodes.data(@node_list, (d) -> d.get('id'))
+      @nodes.data(@nodes_excluding_loss, (d) -> d.get('id'))
         .attr("data-tooltip", (d) =>
           h = ""
           for l in d.right_links

        .attr('id', 'loss_gradient').call(
          (g) ->
            g.append('svg:stop').attr('offset', '0%').attr('style', 'stop-color:rgb(250,0,0);stop-opacity:1')
            g.append('svg:stop').attr('offset', '100%').attr('style', 'stop-color:rgb(250,0,0);stop-opacity:0.1')
      )

    draw_links: =>

    draw_nodes: =>
      colors = d3.scale.category20()
      nodes = @svg.selectAll("g.node")
        .data(@nodes_excluding_loss, (d) -> d.get('id'))
        .enter()
        .append("g")
        .attr("class", "node")
