diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 937d622ec..cd0d9c87a 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -1626,7 +1626,11 @@ async function onModelChange() {
     }
 
     if (oai_settings.chat_completion_source == chat_completion_sources.SCALE) {
-        $('#openai_max_context').attr('max', scale_max);
+        if (oai_settings.max_context_unlocked) {
+            $('#openai_max_context').attr('max', unlocked_max);
+        } else {
+            $('#openai_max_context').attr('max', scale_max);
+        }
         oai_settings.openai_max_context = Math.min(Number($('#openai_max_context').attr('max')), oai_settings.openai_max_context);
         $('#openai_max_context').val(oai_settings.openai_max_context).trigger('input');
     }