diff --git a/tests/installation/agama.pm b/tests/installation/agama.pm
index a64bb0db2194..0be2b2f1d7a7 100644
--- a/tests/installation/agama.pm
+++ b/tests/installation/agama.pm
@@ -131,13 +131,19 @@ sub run {
     assert_and_click('agama-show-tabs');
 
     # default is just a minimal server style install
-    if (check_var('DESKTOP', 'gnome')) {
+    if (get_var('DESKTOP')) {
         assert_and_click('agama-software-tab');
 
-        wait_still_screen 5;
+        wait_still_screen(5);
         assert_and_click('agama-change-software-selection');
-        wait_still_screen 5;
-        assert_and_click('agama-software-selection-gnome-desktop-wayland');
+        wait_still_screen(5);
+
+        if (check_var('DESKTOP', 'gnome')) {
+            assert_and_click('agama-software-selection-gnome-desktop-wayland');
+        } elsif (check_var('DESKTOP', 'kde')) {
+            assert_and_click('agama-software-selection-kde-desktop-wayland');
+        }
+
         assert_and_click('agama-software-selection-close');
     }