diff --git a/time-series.js b/time-series.js
index 61fd6e394f0cdc74acb9af09548209c410fe3c02..89cd2bd318a0319af79d3c7569b7b7f5fe295487 100644
--- a/time-series.js
+++ b/time-series.js
@@ -83,15 +83,13 @@ d3.csv("schengen_data.csv").then((data) => {
                 });
         }
 
-        const legend = svg.selectAll(".legend-group").data([null]);
-
-        const legendEnter = legend.enter().append("g").attr("class", "legend-group")
-            .attr("transform", `translate(${width - 150}, 20)`);
-
-        const legendItems = Array.from(allStates);
-
-        legendEnter.selectAll(".legend-item")
-            .data(legendItems)
+        const legend = svg
+            .selectAll(".legend-group")
+            .data([null])
+            .enter().append("g").attr("class", "legend-group")
+            .attr("transform", `translate(${width - 150}, 20)`)
+            .selectAll(".legend-item")
+            .data(Array.from(allStates))
             .enter()
             .append("g")
             .attr("class", "legend-item")