diff --git a/src/test/java/com/gxwebsoft/generator/templates/entity.java.btl b/src/test/java/com/gxwebsoft/generator/templates/entity.java.btl
index 2bd35ce..bb9b781 100644
--- a/src/test/java/com/gxwebsoft/generator/templates/entity.java.btl
+++ b/src/test/java/com/gxwebsoft/generator/templates/entity.java.btl
@@ -53,27 +53,22 @@ public class ${entity} implements Serializable {
 <% for(field in table.fields) { 
     var keyPropertyName = field.keyFlag ? field.propertyName : null;
 %>
-
-<% if(isNotEmpty(field.comment)) { %>
-<% if(swagger2) { %>    @Schema(description = "${field.comment}")
+<% if(isNotEmpty(field.comment)) { %><% if(swagger2) { %>    @Schema(description = "${field.comment}")
 <% }else{ %>    /**
      * ${field.comment}
      */
-<% } 
-}%><% /* 为时间类型字段添加日期格式化注解并声明字段 */ %><% if(field.propertyType == 'LocalDateTime' || field.propertyType == 'LocalDate' || field.propertyType == 'LocalTime' || field.propertyType == 'Date') { %><% if(field.propertyType == 'LocalDateTime') { %>    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
+<% } %><% } %><% /* 为时间类型字段添加日期格式化注解 */ %><% if(field.propertyType == 'LocalDateTime' || field.propertyType == 'LocalDate' || field.propertyType == 'LocalTime' || field.propertyType == 'Date') { %><% if(field.propertyType == 'LocalDateTime') { %>    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
 <% } else if(field.propertyType == 'LocalDate') { %>    @JsonFormat(pattern = "yyyy-MM-dd")
 <% } else if(field.propertyType == 'LocalTime') { %>    @JsonFormat(pattern = "HH:mm:ss")
 <% } else if(field.propertyType == 'Date') { %>    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
-<% } %><% } %><% /* 主键 */ %><% if(field.keyFlag) { %><% if(field.keyIdentityFlag) { %>    @TableId(value = "${field.annotationColumnName}", type = IdType.AUTO)
+<% } %><% } %><% /* 添加其他注解（主键、字段填充等） */ %><% if(field.keyFlag) { %><% if(field.keyIdentityFlag) { %>    @TableId(value = "${field.annotationColumnName}", type = IdType.AUTO)
 <% } else if(isNotEmpty(idType)) { %>    @TableId(value = "${field.annotationColumnName}", type = IdType.${idType})
 <% } else if(field.convert) { %>    @TableId("${field.annotationColumnName}")
-<% } %><% /* 普通字段 */ %>
-<% } else if(isNotEmpty(field.fill)) { %><% if(field.convert){ %>    @TableField(value = "${field.annotationColumnName}", fill = FieldFill.${field.fill})
+<% } %><% } else if(isNotEmpty(field.fill)) { %><% if(field.convert){ %>    @TableField(value = "${field.annotationColumnName}", fill = FieldFill.${field.fill})
 <% }else{ %>    @TableField(fill = FieldFill.${field.fill})
-<% } %>
-<% } else if(field.convert) { %>    @TableField("${field.annotationColumnName}")
-<% } %><% /* 乐观锁注解 */ %><% if(versionFieldName!'' == field.name) { %>    @Version
-<% } %><% /* 逻辑删除注解 */ %><% if(logicDeleteFieldName!'' == field.name) { %>    @TableLogic
+<% } %><% } else if(field.convert) { %>    @TableField("${field.annotationColumnName}")
+<% } %><% if(versionFieldName!'' == field.name) { %>    @Version
+<% } %><% if(logicDeleteFieldName!'' == field.name) { %>    @TableLogic
 <% } %>    private ${field.propertyType} ${field.propertyName};
 <% } %>
 <% /** -----------END 字段循环遍历----------- **/ %>