Index: uspace/lib/ui/test/msgdialog.c
===================================================================
--- uspace/lib/ui/test/msgdialog.c	(revision cde067e3c7553fc8ac92412868940e574bda9a32)
+++ uspace/lib/ui/test/msgdialog.c	(revision cb747b3b787df15184afceecdeef820cfe7e4226)
@@ -1,4 +1,4 @@
 /*
- * Copyright (c) 2021 Jiri Svoboda
+ * Copyright (c) 2024 Jiri Svoboda
  * All rights reserved.
  *
@@ -92,4 +92,5 @@
 	ui_msg_dialog_params_t params;
 	ui_msg_dialog_t *dialog = NULL;
+	unsigned i;
 	test_cb_resp_t resp;
 
@@ -100,4 +101,5 @@
 	params.caption = "Message";
 	params.text = "Hello";
+	params.choice = umdc_ok_cancel;
 
 	rc = ui_msg_dialog_create(ui, &params, &dialog);
@@ -106,17 +108,19 @@
 
 	/* Button callback with no callbacks set */
-	ui_pbutton_clicked(dialog->bok);
+	ui_pbutton_clicked(dialog->btn[0]);
 
 	/* Button callback with callback not implemented */
 	ui_msg_dialog_set_cb(dialog, &dummy_msg_dialog_cb, NULL);
-	ui_pbutton_clicked(dialog->bok);
+	ui_pbutton_clicked(dialog->btn[0]);
 
-	/* Button callback with real callback set */
-	resp.button = false;
-	resp.bnum = 123;
-	ui_msg_dialog_set_cb(dialog, &test_msg_dialog_cb, &resp);
-	ui_pbutton_clicked(dialog->bok);
-	PCUT_ASSERT_TRUE(resp.button);
-	PCUT_ASSERT_INT_EQUALS(0, resp.bnum);
+	for (i = 0; i < 2; i++) {
+		/* Button callback with real callback set */
+		resp.button = false;
+		resp.bnum = 123;
+		ui_msg_dialog_set_cb(dialog, &test_msg_dialog_cb, &resp);
+		ui_pbutton_clicked(dialog->btn[i]);
+		PCUT_ASSERT_TRUE(resp.button);
+		PCUT_ASSERT_INT_EQUALS(i, resp.bnum);
+	}
 
 	ui_msg_dialog_destroy(dialog);