diff --git a/src/components/Button.astro b/src/components/Button.astro
index 9e89d05..57c8014 100644
--- a/src/components/Button.astro
+++ b/src/components/Button.astro
@@ -116,5 +116,5 @@ function setupThemeToggle() {
 
 // Initialize click event (on first load or page transition)
 setupThemeToggle()
-document.addEventListener('astro:page-load', setupThemeToggle)
+document.addEventListener('astro:after-swap', setupThemeToggle)
 </script>