diff --git a/js/dracula_graph.js b/js/dracula_graph.js
index <HASH>..<HASH> 100644
--- a/js/dracula_graph.js
+++ b/js/dracula_graph.js
@@ -70,9 +70,12 @@ Graph.prototype = {
     //this.snapshots.push({comment: comment, graph: graph});
   },
   removeNode: function(id) {
+    this.nodes[id].shape.remove();
     delete this.nodes[id];
     for(var i = 0; i < this.edges.length; i++) {
       if (this.edges[i].source.id == id || this.edges[i].target.id == id) {
+        this.edges[i].connection.fg.remove();
+        this.edges[i].connection.label.remove();
         this.edges.splice(i, 1);
         i--;
       }
updated `Graph.removeNode` to correctly remove all the connected shapes