diff --git a/src/shell.rs b/src/shell.rs
index 38f0388..566b64b 100644
--- a/src/shell.rs
+++ b/src/shell.rs
@@ -74,7 +74,11 @@ mod tests {
     ) {
         let opts = Opts { cmd, hook, echo, resolve_symlinks };
         let source = Bash(&opts).render().unwrap();
-        assert!(source.contains("[[ -n ${__zoxide_result} ]] || return"));
+        if cmd.is_some() {
+            assert!(source.contains("[[ -n ${__zoxide_result} ]] || return"));
+        } else {
+            assert!(!source.contains("[[ -n ${__zoxide_result} ]] || return"));
+        }
     }
 
     #[apply(opts)]