diff --git a/ruoyi/src/main/resources/vm/java/queryBo.java.vm b/ruoyi/src/main/resources/vm/java/queryBo.java.vm
index 865ede4a1..b98b42fc3 100644
--- a/ruoyi/src/main/resources/vm/java/queryBo.java.vm
+++ b/ruoyi/src/main/resources/vm/java/queryBo.java.vm
@@ -12,6 +12,11 @@ import java.util.HashMap;
 #foreach ($import in $importList)
 import ${import};
 #end
+#if($table.crud || $table.sub)
+import com.ruoyi.common.core.domain.BaseEntity;
+#elseif($table.tree)
+import com.ruoyi.common.core.domain.TreeEntity;
+#end
 
 /**
  * ${functionName}分页查询对象 ${tableName}