diff --git a/src/test/java/com/flowingcode/vaadin/addons/orgchart/EditChartDemo.java b/src/test/java/com/flowingcode/vaadin/addons/orgchart/EditChartDemo.java
index 7012a2f..db5d792 100644
--- a/src/test/java/com/flowingcode/vaadin/addons/orgchart/EditChartDemo.java
+++ b/src/test/java/com/flowingcode/vaadin/addons/orgchart/EditChartDemo.java
@@ -212,7 +212,10 @@ private VerticalLayout getEditionLayout() {
      // #endif
     // show-source typeSelector.setItems("Parent(root)", "Child", "Sibling");
     typeSelector.setValue("Child");
-    typeSelector.addThemeVariants(RadioGroupVariant.LUMO_VERTICAL);
+    // #if vaadin eq 0
+    typeSelector.getThemeNames().add("vertical");
+    // #endif
+    // show-source typeSelector.addThemeVariants(RadioGroupVariant.LUMO_VERTICAL);
 
     // New node(s) layout (dynamic)
     newNodeFieldsLayout = createVerticalLayout();
@@ -459,8 +462,11 @@ private void clearSelectedNode() {
 
   private Button createIconButton(VaadinIcon icon) {
     Button iconButton = new Button(icon.create());
-    iconButton.addThemeVariants(ButtonVariant.LUMO_ICON, ButtonVariant.LUMO_SMALL,
-        ButtonVariant.LUMO_TERTIARY_INLINE);
+    // #if vaadin eq 0
+    iconButton.getThemeNames().addAll(Arrays.asList("icon", "tertiary-inline", "small"));
+    // #endif
+    // show-source iconButton.addThemeVariants(ButtonVariant.LUMO_ICON, ButtonVariant.LUMO_SMALL,
+    // show-source    ButtonVariant.LUMO_TERTIARY_INLINE);
     return iconButton;
   }