diff --git a/tests/desktop_upgrade_graphical.pm b/tests/desktop_upgrade_graphical.pm
index 23b74f6f..1cb2678f 100644
--- a/tests/desktop_upgrade_graphical.pm
+++ b/tests/desktop_upgrade_graphical.pm
@@ -65,8 +65,9 @@ sub run {
         if match_has_tag('desktop_package_tool_password'){
             type_very_safely(get_var('USER_PASSWORD','weakpassword'));
             assert_and_click('desktop_package_tool_authenticate','',30);
-        } 
+        }
 
+        assert_and_click('desktop_package_tool_restart','',30); 
 
         boot_to_login_screen;
     }