diff --git a/extensions-builtin/sd_theme_editor/javascript/ui_theme.js b/extensions-builtin/sd_theme_editor/javascript/ui_theme.js
index 3d2f43a1..9a42527d 100644
--- a/extensions-builtin/sd_theme_editor/javascript/ui_theme.js
+++ b/extensions-builtin/sd_theme_editor/javascript/ui_theme.js
@@ -254,7 +254,7 @@ function initTheme() {
 
 
 	const current_style = gradioApp().querySelector('.gradio-container > style');
-	console.log(current_style);
+	//console.log(current_style);
 	//const head = document.head;		
 	//head.appendChild(current_style);
 
diff --git a/javascript/progressbar.js b/javascript/progressbar.js
index d96f2be5..19463cde 100644
--- a/javascript/progressbar.js
+++ b/javascript/progressbar.js
@@ -93,9 +93,19 @@ function requestProgress(id_task, progressbarContainer, gallery, atEnd, onProgre
 
     var removeProgressBar = function(){
         setTitle("")
-        parentProgressbar.removeChild(divProgress)
-        if(parentGallery) parentGallery.removeChild(livePreview)
-        atEnd()
+		if(divProgress){
+			parentProgressbar.removeChild(divProgress)
+		}
+		
+		if(progressbarContainer.id == "txt2img_gallery_container"){			
+			showSubmitButtons('txt2img', true);
+		}else if(progressbarContainer.id == "img2img_gallery_container"){
+			showSubmitButtons('img2img', true);
+		}
+		if(livePreview){
+			if(parentGallery) parentGallery.removeChild(livePreview)
+		}
+		atEnd()
     }
 
     var fun = function(id_task, id_live_preview){