diff --git a/inject.js b/inject.js
index 59d4e3c..1ce2118 100644
--- a/inject.js
+++ b/inject.js
@@ -269,9 +269,9 @@ chrome.extension.sendMessage({}, function(response) {
               let id = node.dataset['vscid'];
               let ctrl = document.querySelector(`div[data-vscid="${id}"]`)
               if (ctrl) {
-                node.classList.remove('vsc-initialized');
                 ctrl.remove();
               }
+              node.classList.remove('vsc-initialized');
               delete node.dataset['vscid'];
             }
           }