diff --git a/mybatis-flex-codegen/src/main/resources/templates/enjoy/entityOrBase.tpl b/mybatis-flex-codegen/src/main/resources/templates/enjoy/entityOrBase.tpl
index 58ce9887..ba4c7b78 100644
--- a/mybatis-flex-codegen/src/main/resources/templates/enjoy/entityOrBase.tpl
+++ b/mybatis-flex-codegen/src/main/resources/templates/enjoy/entityOrBase.tpl
@@ -60,8 +60,7 @@ import java.io.Serial;
 #if(withSwagger && swaggerVersion.getName() == "DOC")
 @Schema(description = "#(table.getComment())")
 #end
-#(table.buildTableAnnotation())
-#end
+#(table.buildTableAnnotation()) #end
 public class #(entityClassName)#if(withActiveRecord) extends Model<#(entityClassName)>#else#(table.buildExtends())#(table.buildImplements())#end  {
 
     #if(jdkVersion >= 14)