diff --git a/ui/index.html b/ui/index.html
index 6f64317e..5ef9a012 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -32,7 +32,7 @@
             <div id="server-status-color">●</div>
             <span id="server-status-msg">Stable Diffusion is starting..</span>
         </div>
-        <div class="tab-container">
+        <div id="tab-container" class="tab-container">
             <span id="tab-main" class="tab active">
                 <span><i class="fa fa-image icon"></i> Generate</span>
             </span>