diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 938100ba..1757a7e8 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -1088,10 +1088,6 @@ useUpscalingField.addEventListener('change', function(e) {
     upscaleModelField.disabled = !this.checked
 })
 
-if (useBetaChannelField.checked) {
-    updateBranchLabel.innerText = "(beta)"
-}
-
 makeImageBtn.addEventListener('click', makeImage)
 
 document.onkeydown = function(e) {
diff --git a/ui/media/js/parameters.js b/ui/media/js/parameters.js
index 394e9703..d592576d 100644
--- a/ui/media/js/parameters.js
+++ b/ui/media/js/parameters.js
@@ -205,6 +205,7 @@ async function getAppConfig() {
 
         if (config.update_branch === 'beta') {
             useBetaChannelField.checked = true
+			document.querySelector("#updateBranchLabel").innerText = "(beta)"
         }
         if (config.ui && config.ui.open_browser_on_start === false) {
             uiOpenBrowserOnStartField.checked = false