diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx
index 987b6cd911..8e92c37923 100644
--- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx
@@ -30,7 +30,6 @@ export const CurrentImageButtons = memo(() => {
   const ctx = useImageViewerContext();
   const hasProgressImage = useStore(ctx.$hasProgressImage);
   const shouldShowProgressInViewer = useAppSelector(selectShouldShowProgressInViewer);
-
   const isDisabledOverride = hasProgressImage && shouldShowProgressInViewer;
 
   const imageName = useAppSelector(selectLastSelectedImage);
diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx
index 1b4ce009f2..c57781b207 100644
--- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx
@@ -88,6 +88,7 @@ export const CurrentImagePreview = memo(({ imageDTO }: { imageDTO: ImageDTO | nu
       return;
     }
     if (progressEvent.session_id === imageDTO.session_id) {
+      setProgressEvent(null);
       setProgressImage(null);
     }
   }, [imageDTO, progressEvent]);