diff --git a/templates/control.html b/templates/control.html
index 60db72a..46fdf72 100644
--- a/templates/control.html
+++ b/templates/control.html
@@ -452,7 +452,8 @@
                 buttonEl.textContent = `PRODUCTION ON PU ${puNumber}`;
                 buttonEl.disabled = true;
             };
-            } else if (currentState === 'PRODUCTION') {
+            }
+            else if (currentState === 'SYSTEM_MODE_PRODUCTION') {
                     // ✅ Directly update if already in production
                     buttonEl.classList.remove('in-progress');
                     buttonEl.classList.add('production');