From 194fcc107680f8d84ca75749c375c152f991eedd Mon Sep 17 00:00:00 2001
From: Roly Fentanes <roly426@gmail.com>
Date: Sun, 1 Oct 2017 15:29:59 -0400
Subject: [PATCH] Also remove className regardless of ctrl

---
 inject.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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'];
             }
           }