From 57ae8d9c0c51c6e1ee22dde841f3771f487aab1f Mon Sep 17 00:00:00 2001
From: Lawrence Chen <54008264+lawrencecchen@users.noreply.github.com>
Date: Wed, 28 Jan 2026 03:01:24 -0800
Subject: [PATCH] Keep update pill visible during checks

---
 Sources/Update/UpdateDriver.swift | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/Sources/Update/UpdateDriver.swift b/Sources/Update/UpdateDriver.swift
index 42ac4a6c..e91b7067 100644
--- a/Sources/Update/UpdateDriver.swift
+++ b/Sources/Update/UpdateDriver.swift
@@ -196,6 +196,16 @@ class UpdateDriver: NSObject, SPUUserDriver {
             standard.dismissUpdateInstallation()
             return
         }
+        if case .notFound = viewModel.state {
+            UpdateLogStore.shared.append("dismiss update installation ignored (notFound visible)")
+            standard.dismissUpdateInstallation()
+            return
+        }
+        if case .checking = viewModel.state {
+            UpdateLogStore.shared.append("dismiss update installation ignored (checking)")
+            standard.dismissUpdateInstallation()
+            return
+        }
         setState(.idle)
         standard.dismissUpdateInstallation()
     }