1 files changed, 3 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs
index f3d3846..f691b6f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -80,7 +80,9 @@ async fn main() -> Result<()> {
     if cli.dry_run {
         config.write().dry_run = true;
     }
-    if let Some(name) = &cli.role {
+    if let Some(prompt) = &cli.prompt {
+        config.write().set_prompt(prompt)?;
+    } else if let Some(name) = &cli.role {
         config.write().set_role(name)?;
     } else if cli.execute {
         config.write().set_role(SHELL_ROLE)?;