diff --git a/needles/kde/desktop_package_tool_action_select-kde-20241005.json b/needles/kde/desktop_package_tool_action_select-kde-20241005.json
new file mode 100644
index 00000000..e9262594
--- /dev/null
+++ b/needles/kde/desktop_package_tool_action_select-kde-20241005.json
@@ -0,0 +1,15 @@
+{
+  "area": [
+    {
+      "type": "match",
+      "height": 15,
+      "xpos": 501,
+      "width": 99,
+      "ypos": 690
+    }
+  ],
+  "properties": [],
+  "tags": [
+    "desktop_package_tool_action_select"
+  ]
+}
\ No newline at end of file
diff --git a/needles/kde/desktop_package_tool_action_select-kde-20241005.png b/needles/kde/desktop_package_tool_action_select-kde-20241005.png
new file mode 100644
index 00000000..383fee85
Binary files /dev/null and b/needles/kde/desktop_package_tool_action_select-kde-20241005.png differ
diff --git a/needles/kde/desktop_package_tool_restart_automatically-kde-20241005.json b/needles/kde/desktop_package_tool_restart_automatically-kde-20241005.json
new file mode 100644
index 00000000..41cb336a
--- /dev/null
+++ b/needles/kde/desktop_package_tool_restart_automatically-kde-20241005.json
@@ -0,0 +1,15 @@
+{
+  "area": [
+    {
+      "xpos": 499,
+      "width": 49,
+      "ypos": 629,
+      "type": "match",
+      "height": 16
+    }
+  ],
+  "properties": [],
+  "tags": [
+    "desktop_package_tool_restart_automatically"
+  ]
+}
\ No newline at end of file
diff --git a/needles/kde/desktop_package_tool_restart_automatically-kde-20241005.png b/needles/kde/desktop_package_tool_restart_automatically-kde-20241005.png
new file mode 100644
index 00000000..56116cb0
Binary files /dev/null and b/needles/kde/desktop_package_tool_restart_automatically-kde-20241005.png differ
diff --git a/tests/graphical_upgrade_run.pm b/tests/graphical_upgrade_run.pm
index 947b6689..499044b4 100644
--- a/tests/graphical_upgrade_run.pm
+++ b/tests/graphical_upgrade_run.pm
@@ -80,6 +80,8 @@ sub run {
         assert_and_click("desktop_package_tool_update_apply", timeout => 1200);
         # Once we click that button, we can check the checkbutton
         # for restarting the computer automatically.
+        assert_screen ["desktop_package_tool_restart_automatically", "desktop_package_tool_action_select"];
+        click_lastmatch if (match_has_tag "desktop_package_tool_action_select");
         assert_and_click("desktop_package_tool_restart_automatically");
         # When we see auth_required, it means the restart has been triggered
         # and we need to authorize it