diff --git a/tests/server_cockpit_updates.pm b/tests/server_cockpit_updates.pm
index 40a209fc..0ae85597 100644
--- a/tests/server_cockpit_updates.pm
+++ b/tests/server_cockpit_updates.pm
@@ -46,18 +46,19 @@ sub run {
         # and relog for further interaction. We will check if reconnection is
         # needed and if so, we will restart Firefox and login again.
         #
+        break if (check_screen("cockpit_updates_updated"));
         if (check_screen("cockpit_updates_reconnect", 1)) {
             quit_firefox;
             sleep 5;
             start_cockpit(1);
             select_cockpit_update();
-            $run = 50;
+            break;
 
         }
         # Ignore rebooting the system because we want to finish the test instead.
         elsif (check_screen('cockpit_updates_restart_ignore', 1)) {
             assert_and_click 'cockpit_updates_restart_ignore';
-            $run = 50;
+            break;
         }
         else {
             sleep 10;