1 files changed, 8 insertions, 2 deletions
diff --git a/javascript/ui.js b/javascript/ui.js
index e2b9bfe4..b63b84b2 100644
--- a/javascript/ui.js
+++ b/javascript/ui.js
@@ -208,17 +208,23 @@ function submit_img2img(){
 }
 
 function restoreProgressTxt2img(){
+    showRestoreProgressButton("txt2img", false)
     var id = localStorage.getItem("txt2img_task_id")
 
+    id = localStorage.getItem("txt2img_task_id")
+
     if(id) {
         requestProgress(id, gradioApp().getElementById('txt2img_gallery_container'), gradioApp().getElementById('txt2img_gallery'), function(){
             showSubmitButtons('txt2img', true)
         }, null, 0)
     }
 
-    return [id]
+    return id
 }
+
 function restoreProgressImg2img(){
+    showRestoreProgressButton("img2img", false)
+    
     var id = localStorage.getItem("img2img_task_id")
 
     if(id) {
@@ -227,7 +233,7 @@ function restoreProgressImg2img(){
         }, null, 0)
     }
 
-    return [id]
+    return id
 }