diff --git a/vcs/jj/config/sh/alias.d/jj.sh b/vcs/jj/config/sh/alias.d/jj.sh
index 9bf1ff5..8e1c841 100644
--- a/vcs/jj/config/sh/alias.d/jj.sh
+++ b/vcs/jj/config/sh/alias.d/jj.sh
@@ -64,7 +64,9 @@ alias jrb="jj rebase"
 
 # 'branching' bookmark work
 alias jb="jj bookmark"
-alias jbm="jj bookmark set main"
+jbm() {
+    jj bookmark set -r "${1:-@}" main
+}
 
 # remote work
 alias jrv="jj git remote list"