diff --git a/pikit-web/assets/diaglog.js b/pikit-web/assets/diaglog.js
index c10da86..293e8d3 100644
--- a/pikit-web/assets/diaglog.js
+++ b/pikit-web/assets/diaglog.js
@@ -131,6 +131,8 @@ export async function initDiagUI({ elements, toast }) {
     } finally {
       setBusy(false);
     }
+    if (logButton) logButton.classList.toggle("hidden", !uiEnabled);
+    if (!uiEnabled && modal) modal.classList.add("hidden");
   });
 
   debugToggle?.addEventListener("change", async () => {