diff --git a/pigx-visual/pigx-codegen/src/main/resources/template/Entity.java.vm b/pigx-visual/pigx-codegen/src/main/resources/template/Entity.java.vm
index 64ba13d41..8305a6aa7 100755
--- a/pigx-visual/pigx-codegen/src/main/resources/template/Entity.java.vm
+++ b/pigx-visual/pigx-codegen/src/main/resources/template/Entity.java.vm
@@ -17,9 +17,8 @@
 
 package ${package}.${moduleName}.entity;
 
-#if($dbType != "mysql")
+
 import com.baomidou.mybatisplus.annotation.IdType;
-#end
 import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
 import com.baomidou.mybatisplus.extension.activerecord.Model;
@@ -52,11 +51,7 @@ public class ${className} extends Model<${className}> {
      * $column.comments
      */
 #if($column.columnName == $pk.columnName)
-#if($dbType == "mysql")
-    @TableId
-#else
     @TableId(type = IdType.ASSIGN_ID)
-#end
 #end
     @ApiModelProperty(value="$column.comments"#if($column.hidden),hidden=$column.hidden#end)
     private $column.attrType $column.lowerAttrName;