diff --git a/controller/templates/index.html b/controller/templates/index.html
index c51b99b7..0308c52e 100644
--- a/controller/templates/index.html
+++ b/controller/templates/index.html
@@ -2404,6 +2404,7 @@
             if (upgradeBanner) {
                 upgradeBanner.classList.add('show');
                 document.getElementById('script-running-name').textContent = 'WROLPi Upgrade Running';
+                currentScriptOutput = 'upgrade';
             }
 
             const upgradeIsDocker = {{ 'true' if docker_mode else 'false' }};