diff --git a/code.js b/code.js
index 9712e1e..14f27a6 100644
--- a/code.js
+++ b/code.js
@@ -387,6 +387,10 @@ $(function () { // on dom ready
 				cy.on('select', 'node', function(event){
 			    	selectedForEditNodes = cy.$('node:selected');
 				});
+				
+				cy.on('unselect', 'node', function(event){
+			    	selectedForEditNodes = cy.$('node:selected');
+				});
 			},
 
   			// initial viewport state: