diff --git a/designer-base/src/main/java/com/fr/design/mainframe/widget/wrappers/TreeModelWrapper.java b/designer-base/src/main/java/com/fr/design/mainframe/widget/wrappers/TreeModelWrapper.java
index 8a2489407c..00171f3fb0 100644
--- a/designer-base/src/main/java/com/fr/design/mainframe/widget/wrappers/TreeModelWrapper.java
+++ b/designer-base/src/main/java/com/fr/design/mainframe/widget/wrappers/TreeModelWrapper.java
@@ -6,8 +6,7 @@ import com.fr.data.impl.TreeNodeWrapper;
 import com.fr.design.Exception.ValidationException;
 import com.fr.design.designer.properties.Decoder;
 import com.fr.design.designer.properties.Encoder;
-
-import com.fr.general.NameObject;
+import com.fr.form.ui.tree.LayerConfig;
 import com.fr.stable.StringUtils;
 
 public class TreeModelWrapper implements Encoder, Decoder {
@@ -22,7 +21,7 @@ public class TreeModelWrapper implements Encoder, Decoder {
         } else if (v instanceof TreeNodeWrapper) {
             TreeNodeAttr[] treeNodeAttrs = ((TreeNodeWrapper) v).getTreeNodeAttrs();
             return TemplateUtils.render(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Total_N_Grade"), new String[]{"N"}, new String[]{treeNodeAttrs.length + ""});
-        } else if (v instanceof NameObject) {
+        } else if (v instanceof LayerConfig[]) {
             return com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_DataTable_Build");
         } else {
             return com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Auto_Build");