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