diff --git a/ui/media/main.js b/ui/media/main.js
index 8c28998a..ab7f82a2 100644
--- a/ui/media/main.js
+++ b/ui/media/main.js
@@ -669,7 +669,7 @@ async function checkTasks() {
         let success = await doMakeImage(newTask)
         task.batchesDone++
 
-        if (!newTask.isProcessing) {
+        if (!newTask.isProcessing || !success) {
             task.isProcessing = false
             break
         }