diff --git a/modules/ui.py b/modules/ui.py
index 1c9b0f37b..e0ed26d95 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -197,7 +197,7 @@ def check_progress_call(id_part):
 
         image = shared.state.current_image
 
-        if image is None or progress >= 1:
+        if image is None:
             image = gr.update(value=None)
         else:
             preview_visibility = gr_show(True)