diff --git a/tests/ui-ij-tests/src/test/kotlin/ui/UiTests.kt b/tests/ui-ij-tests/src/test/kotlin/ui/UiTests.kt
index b54d16357..1f5c3d133 100644
--- a/tests/ui-ij-tests/src/test/kotlin/ui/UiTests.kt
+++ b/tests/ui-ij-tests/src/test/kotlin/ui/UiTests.kt
@@ -291,7 +291,7 @@ class UiTests {
     waitFor {
       val generateDialog = findAll<ComponentFixture>(byXpath("//div[@class='EngravedLabel']"))
       if (generateDialog.size == 1) {
-        return@waitFor generateDialog.single().hasText("Path With Line Number")
+        return@waitFor hasText("Path With Line Number")
       }
       return@waitFor false
     }