diff --git a/cli/src/commands/add.rs b/cli/src/commands/add.rs
index a9001737..4832470f 100644
--- a/cli/src/commands/add.rs
+++ b/cli/src/commands/add.rs
@@ -1,4 +1,4 @@
-use std::{collections::HashSet, path::PathBuf};
+use std::{collections::HashSet, path::Path};
 
 use sqlx::{types::chrono::Utc, PgConnection};
 use tokio::fs;
@@ -10,7 +10,7 @@ use crate::{
 
 pub async fn add(
     payload: &Payload,
-    output_path: &PathBuf,
+    output_path: &Path,
     mut conn: PgConnection,
 ) -> anyhow::Result<()> {
     let existing_versions = extension_versions(&mut conn, &payload.metadata.extension_name).await?;
@@ -21,7 +21,7 @@ pub async fn add(
     let mut migration_content = String::new();
 
     // Header with metadata
-    migration_content.push_str("-- Migration generated by dbdev add at:");
+    migration_content.push_str("-- Migration generated by dbdev add at: ");
     migration_content.push_str(&Utc::now().format("%Y-%m-%d %H:%M:%S").to_string());
     migration_content.push('\n');
 
@@ -29,13 +29,13 @@ pub async fn add(
     migration_content.push_str(&payload.metadata.extension_name);
     migration_content.push('\n');
 
-    migration_content.push_str("-- Default version:");
+    migration_content.push_str("-- Default version: ");
     migration_content.push_str(&payload.metadata.default_version);
     migration_content.push('\n');
 
     if let Some(comment) = &payload.metadata.comment {
-        migration_content.push_str("-- Comment:");
-        migration_content.push_str(&comment);
+        migration_content.push_str("-- Comment: ");
+        migration_content.push_str(comment);
         migration_content.push('\n');
     }
 
@@ -69,7 +69,7 @@ pub async fn add(
                 versions_installed_now.insert(install_file.version.clone());
             } else {
                 // For initial installation
-                migration_content.push_str("-- Initial installation of version:");
+                migration_content.push_str("-- Initial installation of version: ");
                 migration_content.push_str(&install_file.version);
                 migration_content.push('\n');
 
@@ -86,7 +86,7 @@ pub async fn add(
                 migration_content.push_str("', '");
                 migration_content.push_str(&install_file.version);
                 migration_content.push_str("', $COMMENT$");
-                migration_content.push_str(&payload.metadata.comment.as_deref().unwrap_or(""));
+                migration_content.push_str(payload.metadata.comment.as_deref().unwrap_or(""));
                 migration_content.push_str("$COMMENT$, $SQL$");
                 migration_content.push_str(&install_file.body);
                 migration_content.push_str("$SQL$, ARRAY[");
@@ -100,10 +100,7 @@ pub async fn add(
     }
 
     let existing_update_paths =
-        match update_paths(&mut conn, &payload.metadata.extension_name).await {
-            Ok(paths) => paths,
-            Err(_) => HashSet::new(),
-        };
+        (update_paths(&mut conn, &payload.metadata.extension_name).await).unwrap_or_default();
 
     for upgrade_file in &payload.upgrade_files {
         let update_path = UpdatePath {
diff --git a/cli/src/main.rs b/cli/src/main.rs
index 22be499f..1ca7f674 100644
--- a/cli/src/main.rs
+++ b/cli/src/main.rs
@@ -166,7 +166,7 @@ async fn main() -> anyhow::Result<()> {
             let payload = models::Payload::from_path(extension_dir)?;
             let conn = util::get_connection(connection).await?;
 
-            commands::add::add(&payload, &output_path, conn).await?;
+            commands::add::add(&payload, output_path, conn).await?;
 
             Ok(())
         }