diff --git a/tests/mysql/generator_template_test.go b/tests/mysql/generator_template_test.go
index d6c3476..420a9ec 100644
--- a/tests/mysql/generator_template_test.go
+++ b/tests/mysql/generator_template_test.go
@@ -411,11 +411,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