MLHR-739 #resolve
diff --git a/front/js/app/lib/widgets/DagWidget/DagWidget.js b/front/js/app/lib/widgets/DagWidget/DagWidget.js
index 748bd2a..466d498 100644
--- a/front/js/app/lib/widgets/DagWidget/DagWidget.js
+++ b/front/js/app/lib/widgets/DagWidget/DagWidget.js
@@ -17,6 +17,7 @@
 var BaseView = DT.widgets.Widget;
 var d3 = require('d3');
 var dagreD3 = require('dagre-d3');
+var settings = DT.settings;
 
 /**
  * Base class for DAG widgets
@@ -385,7 +386,7 @@
         event.preventDefault();
 
         var toggleLocalityLink = this.$el.find('.toggle-locality');
-        var legend = this.$el.find('.logical-dag-legend');
+        var legend = this.$el.find('.dag-legend');
 
         this.showLocality = !this.showLocality;