diff --git a/modules/cli/shell/zsh.nix b/modules/cli/shell/zsh.nix
index 63472f4..6921816 100644
--- a/modules/cli/shell/zsh.nix
+++ b/modules/cli/shell/zsh.nix
@@ -44,7 +44,7 @@ lib.my.makeSwitch {
               jj = {
                 ignore_timeout = true;
                 description = "The current jj status";
-                detect_folders = [ ".jj" ];
+                when = "jj root";
                 symbol = " ";
                 command = ''
                   jj log --revisions @ --no-graph --ignore-working-copy --color always --limit 1 --template '