diff --git a/ui/index.html b/ui/index.html
index eb6465cd..5d2c770b 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -669,7 +669,9 @@ async function doMakeImage(reqBody, batchCount) {
                 try {
                     let stepUpdate = JSON.parse(jsonStr)
 
-                    if (stepUpdate.step !== undefined) {
+                    if (stepUpdate.step === undefined) {
+                        finalJSON += jsonStr
+                    } else {
                         let batchSize = parseInt(reqBody['num_inference_steps'])
                         let overallStepCount = stepUpdate.step + batchesDone * batchSize
                         let totalSteps = batchCount * batchSize