diff --git a/ui/desktop/tests/acceptance/projects/targets-test.js b/ui/desktop/tests/acceptance/projects/targets-test.js
index ebda82cc1..70574f123 100644
--- a/ui/desktop/tests/acceptance/projects/targets-test.js
+++ b/ui/desktop/tests/acceptance/projects/targets-test.js
@@ -227,11 +227,11 @@ module('Acceptance | projects | targets', function (hooks) {
         'tbody tr:first-child td:last-child button',
         'Activate connect mode'
       );
-      assert.ok(find('.rose-dialog-success'), 'Success dialog');
+      assert.ok(find('.connection-dialog'), 'Success dialog');
       assert.equal(findAll('.rose-dialog-footer button').length, 1);
       assert.equal(
         find('.rose-dialog-footer button').textContent.trim(),
-        'OK',
+        'Close',
         'Cannot retry'
       );
       assert.equal(
diff --git a/ui/desktop/tests/acceptance/projects/targets/hosts-test.js b/ui/desktop/tests/acceptance/projects/targets/hosts-test.js
index de1d7162c..326832ff6 100644
--- a/ui/desktop/tests/acceptance/projects/targets/hosts-test.js
+++ b/ui/desktop/tests/acceptance/projects/targets/hosts-test.js
@@ -219,10 +219,10 @@ module('Acceptance | projects | targets | hosts', function (hooks) {
         'tbody tr:first-child td:last-child button',
         'Activate connect mode'
       );
-      assert.ok(find('.rose-dialog-success'), 'Success dialog');
+      assert.ok(find('.connection-dialog'), 'Success dialog');
       assert.equal(
-        find('.rose-dialog-footer .rose-button-primary').textContent.trim(),
-        'OK',
+        find('.rose-dialog-footer .rose-button-secondary').textContent.trim(),
+        'Close',
         'Cannot retry'
       );
       assert.equal(
diff --git a/ui/desktop/tests/acceptance/projects/targets/sessions-test.js b/ui/desktop/tests/acceptance/projects/targets/sessions-test.js
index e6031ab88..a4704307b 100644
--- a/ui/desktop/tests/acceptance/projects/targets/sessions-test.js
+++ b/ui/desktop/tests/acceptance/projects/targets/sessions-test.js
@@ -295,10 +295,10 @@ module('Acceptance | projects | targets | sessions', function (hooks) {
     later(async () => {
       run.cancelTimers();
       await click('.rose-layout-page-actions button', 'Activate connect mode');
-      assert.ok(find('.rose-dialog-success'), 'Success dialog');
+      assert.ok(find('.connection-dialog'), 'Success dialog');
       assert.equal(
-        find('.rose-dialog-footer .rose-button-primary').textContent.trim(),
-        'OK',
+        find('.rose-dialog-footer .rose-button-secondary').textContent.trim(),
+        'Close',
         'Cannot retry'
       );
       assert.equal(
@@ -325,7 +325,7 @@ module('Acceptance | projects | targets | sessions', function (hooks) {
       run.cancelTimers();
       instances.session.update('status', 'active');
       await click('.rose-layout-page-actions button', 'Activate connect mode');
-      assert.ok(find('.rose-dialog-success'), 'Success dialog');
+      assert.ok(find('.connection-dialog'), 'Success dialog');
       await click('.rose-dialog-dismiss');
       assert.equal(
         find(
diff --git a/ui/desktop/tests/acceptance/scopes-test.js b/ui/desktop/tests/acceptance/scopes-test.js
index cb15c0af9..646bf9f67 100644
--- a/ui/desktop/tests/acceptance/scopes-test.js
+++ b/ui/desktop/tests/acceptance/scopes-test.js
@@ -292,11 +292,11 @@ module('Acceptance | scopes', function (hooks) {
         'tbody tr:first-child td:last-child button',
         'Activate connect mode'
       );
-      assert.ok(find('.rose-dialog-success'), 'Success dialog');
+      assert.ok(find('.connection-dialog'), 'Success dialog');
       assert.equal(findAll('.rose-dialog-footer button').length, 1);
       assert.equal(
         find('.rose-dialog-footer button').textContent.trim(),
-        'OK',
+        'Close',
         'Cannot retry'
       );
     }, 750);