diff --git a/main.js b/main.js
index 30842a2..1bc3a65 100644
--- a/main.js
+++ b/main.js
@@ -38,6 +38,7 @@ function setLockingMechanism(element) {
         container.style.overflow = locked ? "hidden" : "auto"
         container.style.pointerEvents = locked ? "none" : "auto";
         button.innerHTML = locked ? lockedIcon : unlockedIcon
+        updateTheme();
     }
 
     toggle()
@@ -45,7 +46,6 @@ function setLockingMechanism(element) {
     button.addEventListener("click", toggle)
     toolbar.prepend(button)
 
-    updateTheme();
 }
 
 function removeLockingMechanism() {
@@ -91,4 +91,4 @@ module.exports = class LockPDF extends Plugin {
         removeLockingMechanism()
     }
 
-}
\ No newline at end of file
+}