diff --git a/app/model/dbmodel/ps_module_shop.go b/app/model/dbmodel/ps_module_shop.go
deleted file mode 100644
index 67d185b..0000000
--- a/app/model/dbmodel/ps_module_shop.go
+++ /dev/null
@@ -1,31 +0,0 @@
-// Code generated by gorm.io/gen. DO NOT EDIT.
-// Code generated by gorm.io/gen. DO NOT EDIT.
-// Code generated by gorm.io/gen. DO NOT EDIT.
-
-package dbmodel
-
-import "git.ma-al.com/goc_marek/gormcol"
-
-const TableNamePsModuleShop = "ps_module_shop"
-
-// PsModuleShop mapped from table <ps_module_shop>
-type PsModuleShop struct {
-	IDModule     int32 `gorm:"column:id_module;primaryKey" json:"id_module"`
-	IDShop       int32 `gorm:"column:id_shop;primaryKey;index:id_shop,priority:1" json:"id_shop"`
-	EnableDevice bool  `gorm:"column:enable_device;not null;default:7" json:"enable_device"`
-}
-
-// TableName PsModuleShop's table name
-func (*PsModuleShop) TableName() string {
-	return TableNamePsModuleShop
-}
-
-var PsModuleShopCols = struct {
-	IDModule gormcol.Field
-	IDShop gormcol.Field
-	EnableDevice gormcol.Field
-}{
-	IDModule: gormcol.Field{}.Set((&PsModuleShop{}).TableName(), "id_module"),
-	IDShop: gormcol.Field{}.Set((&PsModuleShop{}).TableName(), "id_shop"),
-	EnableDevice: gormcol.Field{}.Set((&PsModuleShop{}).TableName(), "enable_device"),
-}
diff --git a/app/model/dbmodel/ps_reassurance.go b/app/model/dbmodel/ps_reassurance.go
deleted file mode 100644
index 4af7bb8..0000000
--- a/app/model/dbmodel/ps_reassurance.go
+++ /dev/null
@@ -1,31 +0,0 @@
-// Code generated by gorm.io/gen. DO NOT EDIT.
-// Code generated by gorm.io/gen. DO NOT EDIT.
-// Code generated by gorm.io/gen. DO NOT EDIT.
-
-package dbmodel
-
-import "git.ma-al.com/goc_marek/gormcol"
-
-const TableNamePsReassurance = "ps_reassurance"
-
-// PsReassurance mapped from table <ps_reassurance>
-type PsReassurance struct {
-	IDReassurance int32  `gorm:"column:id_reassurance;primaryKey;autoIncrement:true" json:"id_reassurance"`
-	IDShop        int32  `gorm:"column:id_shop;not null" json:"id_shop"`
-	FileName      string `gorm:"column:file_name;not null" json:"file_name"`
-}
-
-// TableName PsReassurance's table name
-func (*PsReassurance) TableName() string {
-	return TableNamePsReassurance
-}
-
-var PsReassuranceCols = struct {
-	IDReassurance gormcol.Field
-	IDShop gormcol.Field
-	FileName gormcol.Field
-}{
-	IDReassurance: gormcol.Field{}.Set((&PsReassurance{}).TableName(), "id_reassurance"),
-	IDShop: gormcol.Field{}.Set((&PsReassurance{}).TableName(), "id_shop"),
-	FileName: gormcol.Field{}.Set((&PsReassurance{}).TableName(), "file_name"),
-}
diff --git a/gen.go b/gen.go
index ef5bf08..3be9783 100644
--- a/gen.go
+++ b/gen.go
@@ -336,6 +336,11 @@ func (m *GormGen) cleanModelContent(content []byte) []byte {
 	for _, line := range lines {
 		trimmed := strings.TrimSpace(line)
 
+		// Skip gorm.io/gen header comments.
+		if strings.Contains(trimmed, "Code generated by gorm.io/gen") {
+			continue
+		}
+
 		if trimmed == "import (" {
 			importStarted = true
 			newLines = append(newLines, line)
@@ -368,5 +373,11 @@ func (m *GormGen) cleanModelContent(content []byte) []byte {
 	result = strings.Join(newLines, "\n")
 	result = strings.ReplaceAll(result, "psCategoryDo", "")
 
+	// Remove leading empty lines.
+	result = strings.TrimLeft(result, "\n")
+
+	// Add gormcol header comment at the top.
+	result = "// Code generated by gormcol. DO NOT EDIT.\n\n" + result
+
 	return []byte(result)
 }
diff --git a/gormcol b/gormcol
index 532251b..6725cad 100755
Binary files a/gormcol and b/gormcol differ