diff --git a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt
index 3fd7f071ed2b..04baaa1a29fc 100644
--- a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt
+++ b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt
@@ -272,8 +272,8 @@ fun Driver.createNewNotebookWithMouse(name: String = "New Notebook", type: Noteb
       waitOneText("${type.typeName} Notebook").click()
 
       keyboard {
-        driver.ui.pasteText(name)
-        should("expect $name in the popup") {
+        waitFor("expect $name in the popup") {
+          driver.ui.pasteText(name)
           getAllTexts().any { it.text == name }
         }
         enter() // submit the popup