diff --git a/main.js b/main.js
index 7d64d92..b4a9899 100644
--- a/main.js
+++ b/main.js
@@ -1241,6 +1241,7 @@
     hiderButton.classList.add("collapse-button")
     hiderButton.setAttribute("type", "button")
     hiderButton.addEventListener("click", event => {
+      let container = document.querySelector('.legend-container');
       container.querySelectorAll(".legend-desc.collapsible").forEach(element => {
         element.classList.toggle("collapsed")
       })