diff --git a/tests/desktop_upgrade_graphical.pm b/tests/desktop_upgrade_graphical.pm
index 4a7194b3..b7fdec68 100644
--- a/tests/desktop_upgrade_graphical.pm
+++ b/tests/desktop_upgrade_graphical.pm
@@ -63,7 +63,7 @@ sub run {
         # logged in (but don't fail if it doesn't as we're not testing
         # that)
         
-        if (match_has_tag('desktop_package_tool_password')){
+        if (check_screen('desktop_package_tool_password')){
             type_very_safely(get_var('USER_PASSWORD','weakpassword'));
             assert_and_click('desktop_package_tool_authenticate','',30);
         }