fix

scroll-tech · Jul 3, 2024 · dec9563 · dec9563
1 parent bce282a
commit dec9563
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/scripts/has-focused-lines.ts b/src/scripts/has-focused-lines.ts
@@ -1,9 +1,11 @@
 document.addEventListener("DOMContentLoaded", function () {
-  const preElements = document.querySelectorAll(".expressive-code")
-  preElements.forEach((pre) => {
-    const hasTargetClass = pre.querySelector(".highlight")
-    if (hasTargetClass) {
-      pre.classList.add("has-focused-lines")
-    }
+  document.addEventListener("astro:page-load", () => {
+    const preElements = document.querySelectorAll(".expressive-code")
+    preElements.forEach((pre) => {
+      const hasTargetClass = pre.querySelector(".highlight")
+      if (hasTargetClass) {
+        pre.classList.add("has-focused-lines")
+      }
+    })
   })
 })