diff --git a/build-tools/arkui_transformer/src/arkui_transformer.ts b/build-tools/arkui_transformer/src/arkui_transformer.ts
index 65a17e6bffc1cb5ba5d0935a435a443effbdff08..f8506e5f2a6c1c0e7fbe1707d606be991bac303c 100644
--- a/build-tools/arkui_transformer/src/arkui_transformer.ts
+++ b/build-tools/arkui_transformer/src/arkui_transformer.ts
@@ -54,9 +54,10 @@ function printResult(source: string, file: ComponentFile) {
 
 function withM3Config(f: string, func: () => void) {
   if (uiconfig.withM3File(f)) {
+    let prevUseMemoM3 = uiconfig.useMemoM3
     uiconfig.useMemoM3 = true;
     func()
-    uiconfig.useMemoM3 = false;
+    uiconfig.useMemoM3 = prevUseMemoM3;
     return;
   }
   func()