diff --git a/src/store/modules/user.ts b/src/store/modules/user.ts
index c8aa2ce3..db62ab55 100644
--- a/src/store/modules/user.ts
+++ b/src/store/modules/user.ts
@@ -69,7 +69,7 @@ export const useUserStore = defineStore("user", () => {
   // remove token
   function resetToken() {
     return new Promise<void>((resolve) => {
-      localStorage.setItem(TOKEN_KEY, "");
+      removeToken();
       resetRouter();
       resolve();
     });
diff --git a/src/views/dashboard/index.vue b/src/views/dashboard/index.vue
index f7d1ce1f..8a940a97 100644
--- a/src/views/dashboard/index.vue
+++ b/src/views/dashboard/index.vue
@@ -188,7 +188,7 @@
 </template>
 
 <script setup lang="ts">
-import WebSocketManager from "@/api/socket";
+import WebSocketManager from "@/utils/socket";
 
 defineOptions({
   name: "Dashboard",
@@ -390,7 +390,7 @@ const getNoticeLevelTag = (type: number) => {
 };
 
 function connectWebSocket() {
-  WebSocketManager.getWebSocketClient("/topic/onlineUserCount", (message) => {
+  WebSocketManager.getOrCreateClient("/topic/onlineUserCount", (message) => {
     onlineUserCount.value = JSON.parse(message);
   });
 }