1 files changed, 7 insertions, 7 deletions
diff --git a/server/src/test/java/com/vaadin/tests/components/TreeTest.java b/server/src/test/java/com/vaadin/tests/components/TreeTest.java
index 13ef0acc49..9224934a4c 100644
--- a/server/src/test/java/com/vaadin/tests/components/TreeTest.java
+++ b/server/src/test/java/com/vaadin/tests/components/TreeTest.java
@@ -3,8 +3,8 @@ package com.vaadin.tests.components;
 import org.junit.Assert;
 import org.junit.Test;
 
-import com.vaadin.data.HierarchyData;
-import com.vaadin.data.provider.InMemoryHierarchicalDataProvider;
+import com.vaadin.data.TreeData;
+import com.vaadin.data.provider.TreeDataProvider;
 import com.vaadin.event.CollapseEvent;
 import com.vaadin.event.CollapseEvent.CollapseListener;
 import com.vaadin.event.ExpandEvent;
@@ -52,12 +52,12 @@ public class TreeTest {
     @Test
     public void event_source_is_tree() {
         Tree<String> tree = new Tree<>();
-        HierarchyData<String> hierarchyData = new HierarchyData<>();
-        hierarchyData.addItem(null, "Foo");
-        hierarchyData.addItem("Foo", "Bar");
-        hierarchyData.addItem("Foo", "Baz");
+        TreeData<String> treeData = new TreeData<>();
+        treeData.addItem(null, "Foo");
+        treeData.addItem("Foo", "Bar");
+        treeData.addItem("Foo", "Baz");
         tree.setDataProvider(
-                new InMemoryHierarchicalDataProvider<>(hierarchyData));
+                new TreeDataProvider<>(treeData));
 
         TreeCollapseExpandListener listener = new TreeCollapseExpandListener(
                 tree);