diff --git a/javascript/token-counters.js b/javascript/token-counters.js
index eeea7a5d2..fee30c694 100644
--- a/javascript/token-counters.js
+++ b/javascript/token-counters.js
@@ -48,6 +48,8 @@ function setupTokenCounting(id, id_counter, id_button) {
     var counter = gradioApp().getElementById(id_counter);
     var textarea = gradioApp().querySelector(`#${id} > label > textarea`);
 
+    if (!prompt || !counter) return;
+
     if (counter.parentElement == prompt.parentElement) {
         return;
     }
@@ -67,6 +69,8 @@ function setupTokenCounting(id, id_counter, id_button) {
 function toggleTokenCountingVisibility(id, id_counter, id_button) {
     var counter = gradioApp().getElementById(id_counter);
 
+    if (!counter) return;
+
     counter.style.display = opts.disable_token_counters ? "none" : "block";
     counter.classList.toggle("token-counter-visible", !opts.disable_token_counters);
 }
diff --git a/javascript/ui.js b/javascript/ui.js
index 20309634f..7f9e5152f 100644
--- a/javascript/ui.js
+++ b/javascript/ui.js
@@ -249,6 +249,7 @@ function setupResolutionPasting(tabname) {
     var width = gradioApp().querySelector(`#${tabname}_width input[type=number]`);
     var height = gradioApp().querySelector(`#${tabname}_height input[type=number]`);
     for (const el of [width, height]) {
+        if (!el) continue;
         el.addEventListener('paste', function(event) {
             var pasteData = event.clipboardData.getData('text/plain');
             var parsed = pasteData.match(/^\s*(\d+)\D+(\d+)\s*$/);
diff --git a/script.js b/script.js
index de1a9000d..484bfef7d 100644
--- a/script.js
+++ b/script.js
@@ -119,7 +119,7 @@ var executedOnLoaded = false;
 
 document.addEventListener("DOMContentLoaded", function() {
     var mutationObserver = new MutationObserver(function(m) {
-        if (!executedOnLoaded && gradioApp().querySelector('#txt2img_prompt')) {
+        if (!executedOnLoaded && get_uiCurrentTabContent()) {
             executedOnLoaded = true;
             executeCallbacks(uiLoadedCallbacks);
         }