diff --git a/qui/tray/updates.py b/qui/tray/updates.py
index 736ba441..65aa268d 100644
--- a/qui/tray/updates.py
+++ b/qui/tray/updates.py
@@ -170,8 +170,8 @@ def domain_removed(self, _submitter, _event, vm, *_args, **_kwargs):
         if vm in self.vms_needing_update:
             self.vms_needing_update.remove(vm)
             self.update_indicator_state()
-        if vm.name in self.obsolete_vms:
-            self.obsolete_vms.remove(vm.name)
+        if vm in self.obsolete_vms:
+            self.obsolete_vms.remove(vm)
             self.update_indicator_state()
 
     def feature_unset(self, vm, event, feature, **_kwargs):