diff --git a/app/electron/init.html b/app/electron/init.html
index 01f55b8a4..d9b66a0d4 100644
--- a/app/electron/init.html
+++ b/app/electron/init.html
@@ -362,7 +362,7 @@
     })
   })
   // 防止内核长时间不用被回收
-  new WebSocket(`${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host}/ws`);
+  new WebSocket(`${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host}/ws?app=siyuan&id=init`);
 </script>
 </body>
 </html>
diff --git a/app/stage/auth.html b/app/stage/auth.html
index 5e9c42900..bebf64075 100644
--- a/app/stage/auth.html
+++ b/app/stage/auth.html
@@ -496,7 +496,7 @@
     })
   }
   // 防止内核长时间不用被回收
-  new WebSocket(`${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host}/ws`);
+  new WebSocket(`${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host}/ws?app=siyuan&id=auth`);
 </script>
 </body>
 </html>