diff --git a/index.html b/index.html
index 69e80a5..6188f2a 100644
--- a/index.html
+++ b/index.html
@@ -87,6 +87,7 @@
       </div>
 
       <div class="header-right">
+        <span id="violet-trial-badge" class="violet-badge hidden"></span>
         <span id="my-username-badge" class="my-badge"></span>
         <button id="logout-btn" class="btn-logout">Exit</button>
       </div>
diff --git a/static/chat.js b/static/chat.js
index 9733705..0bde142 100644
--- a/static/chat.js
+++ b/static/chat.js
@@ -367,6 +367,7 @@ socket.on("ignore_status", (data) => {
 });
 
 function updateVioletBadge() {
+    if (!trialBadge) return;
     if (state.hasAiAccess) {
         trialBadge.classList.add("hidden");
     } else {
@@ -587,7 +588,7 @@ $("sidebar-toggle").onclick = () => {
     sidebar.classList.toggle("open");
 };
 
-$("tab-ai-violet").onclick = () => switchTab("ai-violet");
+// tab-ai-violet is created dynamically when user opens Violet PM
 $("tab-lobby").onclick = () => switchTab("lobby");
 
 $("close-paywall").onclick = () => paywallModal.classList.add("hidden");