1 files changed, 2 insertions, 2 deletions
diff --git a/tests/testbench/com/vaadin/launcher/ApplicationRunnerServlet.java b/tests/testbench/com/vaadin/launcher/ApplicationRunnerServlet.java
index 43d4ad699e..2611546203 100644
--- a/tests/testbench/com/vaadin/launcher/ApplicationRunnerServlet.java
+++ b/tests/testbench/com/vaadin/launcher/ApplicationRunnerServlet.java
@@ -112,10 +112,10 @@ public class ApplicationRunnerServlet extends AbstractApplicationServlet {
             final Class<?> classToRun = getClassToRun();
             if (UI.class.isAssignableFrom(classToRun)) {
                 Application application = new Application();
-                application.addRootProvider(new AbstractRootProvider() {
+                application.addUIProvider(new AbstractRootProvider() {
 
                     @Override
-                    public Class<? extends UI> getRootClass(
+                    public Class<? extends UI> getUIClass(
                             Application application, WrappedRequest request)
                             throws UIRequiresMoreInformationException {
                         return (Class<? extends UI>) classToRun;