diff --git a/migrate_mssql/migrate.go b/migrate_mssql/migrate.go
index 67fc830..b2150cc 100644
--- a/migrate_mssql/migrate.go
+++ b/migrate_mssql/migrate.go
@@ -374,7 +374,7 @@ func getIndexStr(index Index) string {
 func getDataType(fieldType string, fieldMap map[string]string) string {
 	var DataType string
 
-	dataTypeVal, dataTypeOk := fieldMap["driver"]
+	dataTypeVal, dataTypeOk := fieldMap["type"]
 	if dataTypeOk {
 		DataType = dataTypeVal
 		if DataType == "double" {
diff --git a/migrate_mysql/migrate.go b/migrate_mysql/migrate.go
index 410c92b..0ad698f 100644
--- a/migrate_mysql/migrate.go
+++ b/migrate_mysql/migrate.go
@@ -392,7 +392,7 @@ func getIndexStr(index Index) string {
 func getDataType(fieldType string, fieldMap map[string]string) string {
 	var DataType string
 
-	dataTypeVal, dataTypeOk := fieldMap["driver"]
+	dataTypeVal, dataTypeOk := fieldMap["type"]
 	if dataTypeOk {
 		DataType = dataTypeVal
 	} else {
diff --git a/migrate_postgres/migrate.go b/migrate_postgres/migrate.go
index 33964a2..6cef995 100644
--- a/migrate_postgres/migrate.go
+++ b/migrate_postgres/migrate.go
@@ -441,7 +441,7 @@ func getIndexStr(index Index) string {
 func getDataType(fieldType string, fieldMap map[string]string) string {
 	var DataType string
 
-	dataTypeVal, dataTypeOk := fieldMap["driver"]
+	dataTypeVal, dataTypeOk := fieldMap["type"]
 	if dataTypeOk {
 		DataType = dataTypeVal
 		if "tinyint" == DataType {
diff --git a/migrate_sqlite3/migrate.go b/migrate_sqlite3/migrate.go
index f3cd053..555ed58 100644
--- a/migrate_sqlite3/migrate.go
+++ b/migrate_sqlite3/migrate.go
@@ -422,7 +422,7 @@ func getIndexStr(index Index) string {
 func getDataType(fieldType string, fieldMap map[string]string) string {
 	var DataType string
 
-	dataTypeVal, dataTypeOk := fieldMap["driver"]
+	dataTypeVal, dataTypeOk := fieldMap["type"]
 	if dataTypeOk {
 		DataType = dataTypeVal
 	} else {