tests/src/FunctionalJavascript/LinkitDialogCKEditor5Test.php | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/src/FunctionalJavascript/LinkitDialogCKEditor5Test.php b/tests/src/FunctionalJavascript/LinkitDialogCKEditor5Test.php
index e9f6899..91c3d52 100644
--- a/tests/src/FunctionalJavascript/LinkitDialogCKEditor5Test.php
+++ b/tests/src/FunctionalJavascript/LinkitDialogCKEditor5Test.php
@@ -203,6 +203,7 @@ class LinkitDialogCKEditor5Test extends WebDriverTestBase {
     // Accept the first autocomplete suggestion.
     $results[0]->click();
     $link_edit_balloon->pressButton('Save');
+    $this->assertTrue($assert_session->waitForElementRemoved('css', '.ck-button-save'));
     // Assert balloon is still visible, but now it's again the link actions one.
     $this->assertVisibleBalloon('.ck-link-actions');
     // Assert balloon can be closed by clicking elsewhere in the editor.