diff --git a/src/popup/popup.js b/src/popup/popup.js
index f8a5ac334792d68e32ada83d41a3e0977be66bec..a1b4a990a9ce2ebddb074449d7faacb1169f5b51 100644
--- a/src/popup/popup.js
+++ b/src/popup/popup.js
@@ -40,6 +40,7 @@ async function updateConnectionButton() {
 
   button.onclick = async () => {
     await browser.runtime.sendMessage({ action: "toggleAuth" });
+    await updateConnectionButton();
   };
 }
 
@@ -238,6 +239,7 @@ async function handleToggleExtension() {
   const accessToken = await getAccessToken();
   if (!accessToken) return;
   await proceedToggleExtension();
+  await updateConnectionButton();
 }
 
 async function proceedToggleExtension() {