diff --git a/pkg/writers/bun/identity_column_test.go b/pkg/writers/bun/identity_column_test.go
index 67bbb81..281cb40 100644
--- a/pkg/writers/bun/identity_column_test.go
+++ b/pkg/writers/bun/identity_column_test.go
@@ -10,7 +10,7 @@ import (
 func TestBuildBunTag_IdentityColumn(t *testing.T) {
 	mapper := NewTypeMapper("", "")
 
-	build := func(pk bool, identity bool, generation string) *models.Column {
+	build := func(pk, identity bool, generation string) *models.Column {
 		col := models.InitColumn("id", "users", "public")
 		col.Type = "bigint"
 		col.IsPrimaryKey = pk
diff --git a/pkg/writers/gorm/identity_column_test.go b/pkg/writers/gorm/identity_column_test.go
index f8f4ee8..e987718 100644
--- a/pkg/writers/gorm/identity_column_test.go
+++ b/pkg/writers/gorm/identity_column_test.go
@@ -10,7 +10,7 @@ import (
 func TestBuildGormTag_IdentityColumn(t *testing.T) {
 	mapper := NewTypeMapper("")
 
-	build := func(pk bool, identity bool, generation string) *models.Column {
+	build := func(pk, identity bool, generation string) *models.Column {
 		col := models.InitColumn("id", "users", "public")
 		col.Type = "bigint"
 		col.IsPrimaryKey = pk