diff --git a/tests/desktop_update_graphical.pm b/tests/desktop_update_graphical.pm
index 1c69ba9c..638681cb 100644
--- a/tests/desktop_update_graphical.pm
+++ b/tests/desktop_update_graphical.pm
@@ -40,9 +40,9 @@ sub run {
         }
     }
     assert_and_click 'desktop_package_tool_update';
-    # if this is KDE and it had already noticed the notification, we
-    # will already have the apply button at this point
-    unless (check_screen 'desktop_package_tool_update_apply', 5) {
+    # depending on automatic update checks, 'apply' or 'download' may
+    # already be visible at this point, we may not need to refresh
+    unless (check_screen ['desktop_package_tool_update_apply', 'desktop_package_tool_update_download'], 5) {
         # refresh updates
         assert_and_click 'desktop_package_tool_update_refresh', '', 120;
     }