diff --git a/src/plugins/tree-shaking.ts b/src/plugins/tree-shaking.ts
index d3e7cf76e..c0b16674c 100644
--- a/src/plugins/tree-shaking.ts
+++ b/src/plugins/tree-shaking.ts
@@ -31,7 +31,7 @@ export const treeShakingPlugin = ({
               return false
             },
             load(id) {
-              if (id === info.path) return code
+              if (id === info.path) return { code, map: info.map }
             },
           },
         ],