diff --git a/src/main.rs b/src/main.rs
index 7a8aa53..53d63ac 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -88,11 +88,12 @@ pub enum DatabaseType {
 }
 
 async fn generate_rust_from_database(args: &Cli) -> DbSetsFsWriter {
-    let database_type = if args.db_url.starts_with("postgres://") {
-        DatabaseType::Postgres
-    } else {
-        DatabaseType::MySql
-    };
+    let database_type =
+        if args.db_url.starts_with("postgres://") || args.db_url.starts_with("postgresql://") {
+            DatabaseType::Postgres
+        } else {
+            DatabaseType::MySql
+        };
 
     if args.mode == Mode::Dbset && database_type == DatabaseType::MySql {
         panic!("DbSet not currently supported for MySql")