diff --git a/src/Assets/ViteCompiler.php b/src/Assets/ViteCompiler.php
index 9d4b934..dae030d 100644
--- a/src/Assets/ViteCompiler.php
+++ b/src/Assets/ViteCompiler.php
@@ -151,11 +151,14 @@ public function enqueueAssets(): void
             }
 
             add_filter('script_loader_tag', function ($tag, $handle) {
-                if (in_array($handle, ['vite-client', $this->handle], true)) {
-                    $tag = str_replace('<script ', '<script type="module" ', $tag);
-                }
-
-                return $tag;
+              $jsHandles = array_map(
+                fn($asset) => "{$this->handle}-" . pathinfo($asset->path(), PATHINFO_FILENAME),
+                array_filter($this->assets, fn($asset) => str_ends_with($asset->path(), '.js'))
+              );
+              if (in_array($handle, array_merge(['vite-client'], $jsHandles), true)) {
+                $tag = str_replace('<script ', '<script type="module" ', $tag);
+              }
+              return $tag;
             }, 10, 2);
 
             return;