diff --git a/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java b/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java
index 675674e4..95fa8f91 100644
--- a/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java
+++ b/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java
@@ -154,6 +154,7 @@ public class GenerateServiceImpl implements IGenerateService {
         gen.put("functionName", genTable.getFunctionName());
         gen.put("treePrimary", genTable.getTreePrimary());
         gen.put("treeParent", genTable.getTreeParent());
+        gen.put("treeName", genTable.getTreeName());
         gen.put("subTableName", genTable.getSubTableName());
         gen.put("subTableFk", genTable.getSubTableFk());
         maps.put("gen", gen);