diff --git a/content/apply.js b/content/apply.js
index bc5fa4c2..0b331163 100644
--- a/content/apply.js
+++ b/content/apply.js
@@ -362,9 +362,8 @@ const APPLY = (() => {
       .then(() => {
         setTimeout(() => {
           el.remove();
-          if (docHadClass) {
-            document.documentElement.classList.remove(className);
-          } else {
+          document.documentElement.classList.remove(className);
+          if (!docHadClass && document.documentElement.classList.length === 0) {
             document.documentElement.removeAttribute('class');
           }
         });