diff --git a/vertx-vaadin-tests/test-root-context/src/test/java/com/vaadin/flow/uitest/ui/PageIT.java b/vertx-vaadin-tests/test-root-context/src/test/java/com/vaadin/flow/uitest/ui/PageIT.java
index b2446aef..6bcc8131 100644
--- a/vertx-vaadin-tests/test-root-context/src/test/java/com/vaadin/flow/uitest/ui/PageIT.java
+++ b/vertx-vaadin-tests/test-root-context/src/test/java/com/vaadin/flow/uitest/ui/PageIT.java
@@ -88,7 +88,7 @@ public void testReload() {
         input.setValue("foo");
         Assert.assertEquals("foo", input.getPropertyString("value"));
         findElement(By.cssSelector("div#reload")).click();
-        input = $(InputTextElement.class).id("input");
+        input = waitUntil( d -> $(InputTextElement.class).id("input"));
         Assert.assertEquals("", input.getValue());
     }