diff --git a/package-lock.json b/package-lock.json
index ebaa92f..1d8b73f 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "retroify.js",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "retroify.js",
-      "version": "0.2.0",
+      "version": "0.2.1",
       "license": "MIT",
       "devDependencies": {
         "@babel/preset-env": "^7.0.0",
diff --git a/package.json b/package.json
index 8e85bf2..8f2de01 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "retroify.js",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "A library to apply retro 8-bit styling to web pages",
   "repository": {
     "type": "git",
diff --git a/src/retroify.js b/src/retroify.js
index 2595a9e..d985f67 100644
--- a/src/retroify.js
+++ b/src/retroify.js
@@ -1,5 +1,5 @@
 /**
- * retroify.js v0.2.0
+ * retroify.js v0.2.1
  * A library to apply retro 8-bit styling to web pages
  * https://github.com/rbonestell/retroify.js
  * 
@@ -308,7 +308,7 @@
     }
 
     const Retroify = {
-        version: '0.2.0',
+        version: '0.2.1',
         
         /**
          * Check if the environment supports Retroify
@@ -357,8 +357,6 @@
                     element.setAttribute('aria-pressed', 'false');
                 }
 
-                element.setAttribute('data-retroify-root', 'true');
-
                 // Skip applying retroify to blacklist element tags
                 const selector = `*:not(${BLACKLISTED_TAGS.join(', ')})`;
 
@@ -398,6 +396,8 @@
                     throw new Error('Invalid target type. Expected HTMLElement or string selector');
                 }
 
+                rootElement.setAttribute('data-retroify-root', 'true');
+
                 // Apply retroify to the target element
                 applyToTargetElement(rootElement);