diff --git a/package.json b/package.json
index 1eadcb6..b0d48d1 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-mock",
-  "version": "2.1.1",
+  "version": "2.1.2",
   "description": "A mock plugin for vite",
   "main": "dist/index.js",
   "files": [
diff --git a/src/index.ts b/src/index.ts
index bc15ae0..6988ebb 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -20,7 +20,9 @@ export function viteMockServe(opt: ViteMockOptions): Plugin {
     configResolved(resolvedConfig) {
       config = resolvedConfig;
       isDev = config.command === 'serve' && !config.isProduction;
-      needSourcemap = resolvedConfig.isProduction && !!resolvedConfig.build.sourcemap;
+      needSourcemap =
+        resolvedConfig.command === 'serve' ||
+        (resolvedConfig.isProduction && !!resolvedConfig.build.sourcemap);
     },
 
     configureServer: async ({ middlewares }) => {
@@ -28,37 +30,27 @@ export function viteMockServe(opt: ViteMockOptions): Plugin {
       if (!localEnabled) return;
       createMockServer(opt);
 
-      // parse application/x-www-form-urlencoded
-      // middlewares.use(bodyParser.urlencoded({ extended: false }));
-      // parse application/json
-      // middlewares.use(bodyParser.json());
       const middleware = await requestMiddle(opt);
       middlewares.use(middleware);
     },
     async transform(code: string, id: string) {
-      const getMap = () => (needSourcemap ? this.getCombinedSourcemap() : null);
+      const getResult = (content: string) => ({
+        map: needSourcemap ? this.getCombinedSourcemap() : null,
+        code: content,
+      });
 
       if (isDev || !id.endsWith(injectFile)) {
-        return {
-          code,
-          map: getMap(),
-        };
+        return getResult(code);
       }
       const { prodEnabled = true, injectCode = '' } = opt;
       if (!prodEnabled) {
-        return {
-          code,
-          map: getMap(),
-        };
+        return getResult(code);
       }
-      return {
-        code: `
+      return getResult(`
       ${code}
       \n
       ${injectCode}
-      `,
-        map: getMap(),
-      };
+      `);
     },
   };
 }