diff --git a/main.js b/main.js
index 21288ee..6fc6722 100644
--- a/main.js
+++ b/main.js
@@ -1449,7 +1449,7 @@
     ])
   }
 
-  function makeLegend(rows) {
+  function makeLegend(container, rows) {
     let legendContainer = Tag.div({
       classes: "legend-container"
     })
@@ -2080,7 +2080,7 @@
 
     /* LEGEND */
 
-    let legendContainer = makeLegend(rows)
+    let legendContainer = makeLegend(container, rows)
     legendContainer.append(makeLegendFiles(rows))
 
     /* APPEND ALL */