diff --git a/README.md b/README.md
index 6303828..fab440e 100644
--- a/README.md
+++ b/README.md
@@ -7,13 +7,18 @@ Proporciona una manera sencilla de indentificar que se está navegando en una we
 Para integrar esta herramienta en un proyecto, copie el siguiente código y coloquelo en el archivo del index.html de su proyecto.
 
 ```html
-<script src="https://cdn.jsdelivr.net/gh/ogticrd/official-header/main.js" defer></script>
+<script
+  id="official-header"
+  src="https://cdn.jsdelivr.net/gh/ogticrd/official-header/main.js"
+  defer
+></script>
 ```
 
 En caso de desear utilizar el tema dark, basta con cambiar la importación del script por:
 
 ```html
 <script
+  id="official-header"
   src="https://cdn.jsdelivr.net/gh/ogticrd/official-header/main.js"
   defer
   theme="dark"
diff --git a/main.js b/main.js
index e8d4604..f2036f9 100644
--- a/main.js
+++ b/main.js
@@ -1,8 +1,8 @@
 class GovernmentOfficialHeader extends HTMLElement {
   constructor() {
     const theme =
-      (document.currentScript &&
-        document.currentScript.getAttribute("theme")) ||
+      (document.getElementById("official-header") &&
+        document.getElementById("official-header").getAttribute("theme")) ||
       "light";
     super();
     const template = document.createElement("template");