diff --git a/tests/postgres/generator_template_test.go b/tests/postgres/generator_template_test.go
index 52f0c3b..7141842 100644
--- a/tests/postgres/generator_template_test.go
+++ b/tests/postgres/generator_template_test.go
@@ -464,11 +464,11 @@ func TestGeneratorTemplate_SQLBuilder_ChangeColumnTypes(t *testing.T) {
 					UseSQLBuilder(template.DefaultSQLBuilder().
 						UseTable(func(table metadata.Table) template.TableSQLBuilder {
 							return template.DefaultTableSQLBuilder(table).
-								UseColumn(func(column metadata.Column) template.TableSQLBuilderColumn {
-									defaultColumn := template.DefaultTableSQLBuilderColumn(column)
+								UseColumn(func(dialect template.Dialect, column metadata.Column) template.TableSQLBuilderColumn {
+									defaultColumn := template.DefaultTableSQLBuilderColumn(dialect, column)
 
 									if defaultColumn.Name == "ActorID" {
-										defaultColumn.Type = "String"
+										defaultColumn.Type = "ColumnString"
 									}
 
 									return defaultColumn
@@ -494,8 +494,8 @@ func TestGeneratorTemplate_Model_SQLBuilder_SkipColumn(t *testing.T) {
 					UseSQLBuilder(template.DefaultSQLBuilder().
 						UseTable(func(table metadata.Table) template.TableSQLBuilder {
 							return template.DefaultTableSQLBuilder(table).
-								UseColumn(func(column metadata.Column) template.TableSQLBuilderColumn {
-									defaultColumn := template.DefaultTableSQLBuilderColumn(column)
+								UseColumn(func(dialect template.Dialect, column metadata.Column) template.TableSQLBuilderColumn {
+									defaultColumn := template.DefaultTableSQLBuilderColumn(dialect, column)
 
 									if defaultColumn.Name == "FirstName" {
 										defaultColumn.Skip = true
@@ -610,8 +610,8 @@ func TestGeneratorTemplate_Model_SqlBuilder_RenameStructFieldNames(t *testing.T)
 					UseSQLBuilder(template.DefaultSQLBuilder().
 						UseTable(func(table metadata.Table) template.TableSQLBuilder {
 							return template.DefaultTableSQLBuilder(table).
-								UseColumn(func(column metadata.Column) template.TableSQLBuilderColumn {
-									defaultSqlBuilderColumn := template.DefaultTableSQLBuilderColumn(column)
+								UseColumn(func(dialect template.Dialect, column metadata.Column) template.TableSQLBuilderColumn {
+									defaultSqlBuilderColumn := template.DefaultTableSQLBuilderColumn(dialect, column)
 
 									if table.Name == "payment" && column.Name == "amount" {
 										defaultSqlBuilderColumn.Name = "AmountInCents"