diff --git a/frontend/src/components/visualization/CitationVis.vue b/frontend/src/components/visualization/CitationVis.vue
index ad8ab28f..29970056 100644
--- a/frontend/src/components/visualization/CitationVis.vue
+++ b/frontend/src/components/visualization/CitationVis.vue
@@ -279,14 +279,13 @@ export default {
             ? "rgba(255, 255, 255," + com.highlight_opacity + ")"
             : "rgba(0, 100, 100, " + com.base_opacity + ")";
         }
-
-        sigma_instance.graph.nodes().forEach(function (n) {
-          var node = sigma_instance.graph.getNodeFromIndex(n.id);
-          !proteins.has(n.id)
-            ? (node.color = "rgb(0,100,100)")
-            : (node.color = "rgb(255,255,255)");
-        });
       }
+      sigma_instance.graph.nodes().forEach(function (n) {
+        var node = sigma_instance.graph.getNodeFromIndex(n.id);
+        !proteins.has(n.id)
+          ? (node.color = "rgb(0,100,100)")
+          : (node.color = "rgb(255,255,255)");
+      });
       this.$store.commit("assign_highlightedSet", highlighted_edges);
 
       sigma_instance.refresh();