diff --git a/packages/hmr/src/wrapMain.ts b/packages/hmr/src/wrapMain.ts
index 5084f04c..60042fb2 100644
--- a/packages/hmr/src/wrapMain.ts
+++ b/packages/hmr/src/wrapMain.ts
@@ -18,20 +18,14 @@ export function ${initFuncId}(app, locals) {
   ${localsObjId} = locals;
 }
 
-if (import.meta.hot) {
-  import.meta.hot.accept((newMainModule) => {
-    newMainModule.${initFuncId}(${appInstDeafultId}, ${localsObjId});
-    const newMain = newMainModule.${mainFuncId}(${localsObjId});
-    if(${appInstDeafultId}.state !== "idle") {
-      ${appInstDeafultId}.promises.mainExecuted.then(() => {
-        ${appInstDeafultId}.main = newMain;
-        ${appInstDeafultId}.update();
-      })
-    } else {
-      ${appInstDeafultId}.main = newMain;
-      ${appInstDeafultId}.update();
-    }
-  });
-}
+import.meta.hot?.accept(async (newMainModule) => {
+  newMainModule.${initFuncId}(${appInstDeafultId}, ${localsObjId});
+  const newMain = newMainModule.${mainFuncId}(${localsObjId});
+  if(${appInstDeafultId}.state !== "idle") {
+    await ${appInstDeafultId}.promises.mainExecuted;
+  }
+  ${appInstDeafultId}.main = newMain;
+   ${appInstDeafultId}.update();
+});
 `);
 }