diff --git a/cmd/goose/main.go b/cmd/goose/main.go
index d46752c..6e726fb 100644
--- a/cmd/goose/main.go
+++ b/cmd/goose/main.go
@@ -45,13 +45,15 @@ func main() {
 
 	driver, dbstring, command := args[0], args[1], args[2]
 
+	if err := goose.SetDialect(driver); err != nil {
+		log.Fatal(err)
+	}
+
 	switch driver {
-	case "postgres", "mysql", "sqlite3", "redshift", "tidb":
-		if err := goose.SetDialect(driver); err != nil {
-			log.Fatal(err)
-		}
-	default:
-		log.Fatalf("%q driver not supported\n", driver)
+	case "redshift":
+		driver = "postgres"
+	case  "tidb":
+		driver = "mysql"
 	}
 
 	switch dbstring {
@@ -60,14 +62,6 @@ func main() {
 	default:
 	}
 
-	if driver == "redshift" {
-		driver = "postgres"
-	}
-
-	if driver == "tidb" {
-		driver = "mysql"
-	}
-
 	db, err := sql.Open(driver, dbstring)
 	if err != nil {
 		log.Fatalf("-dbstring=%q: %v\n", dbstring, err)