diff --git a/visualization/client/js/controllers/vis-general.js b/visualization/client/js/controllers/vis-general.js
index 25da520c7ce1dbde2bac306d7a50e58dcb0731ce..913f1c6dc9418a436bbaca8ef66a20364dac7624 100644
--- a/visualization/client/js/controllers/vis-general.js
+++ b/visualization/client/js/controllers/vis-general.js
@@ -504,8 +504,8 @@ angular
                         edges.update(edges.get());
 
                         angular.forEach(nodes.get(), function(nodeVal, nodeKey){
-                            if( nodeVal.label in $scope.node_shape_mapping ){
-                                var color = $scope.node_color_mapping[nodeVal.label];
+                            if( nodeVal.label_original in $scope.node_shape_mapping ){
+                                var color = $scope.node_color_mapping[nodeVal.label_original];
                                 nodeVal.color.background = color.background;
                             }
                             else{
@@ -575,8 +575,8 @@ angular
                         nodeVal.color.background = "#cccccc";
                     }
                     else{
-                        if( nodeVal.label in $scope.node_color_mapping ){
-                            var color = $scope.node_color_mapping[nodeVal.label];
+                        if( nodeVal.label_original in $scope.node_color_mapping ){
+                            var color = $scope.node_color_mapping[nodeVal.label_original];
                             nodeVal.color.background = color.background;
                         }
                         else{
@@ -853,8 +853,8 @@ angular
                 edges.update(edges.get());
 
                 angular.forEach(nodes.get(), function(nodeVal, nodeKey){
-                    if( nodeVal.label == node_type_label ){
-                        var color = $scope.node_color_mapping[nodeVal.label];
+                    if( nodeVal.label_original == node_type_label ){
+                        var color = $scope.node_color_mapping[nodeVal.label_original];
                         nodeVal.color.background = color.background;
                     }
                     else{