diff --git a/action.yml b/action.yml
index 72012ef..79a3e2b 100644
--- a/action.yml
+++ b/action.yml
@@ -1,9 +1,6 @@
 name: "Setup Dev Drive"
 description: "A GitHub Action to setup a Windows Dev Drive."
 author: "samypr100"
-branding:
-  icon: hard-drive
-  color: purple
 inputs:
   drive-size:
     description: "Drive Size. Examples: 1GB, 3GB, 10GB."
@@ -25,5 +22,8 @@ inputs:
     default: "false"
 runs:
   using: "node20"
-  main: "dist/main.js"
+  main: "dist/setup/index.js"
   post: "dist/post.js"
+branding:
+  icon: hard-drive
+  color: purple
diff --git a/dist/main.js b/dist/setup/index.js
similarity index 100%
rename from dist/main.js
rename to dist/setup/index.js