diff --git a/shell/shellIntegration-rc.zsh b/shell/shellIntegration-rc.zsh
index 55dec90..55cff4c 100644
--- a/shell/shellIntegration-rc.zsh
+++ b/shell/shellIntegration-rc.zsh
@@ -41,8 +41,8 @@ __is_update_cwd() {
 
 __is_update_prompt() {
 	__is_prior_prompt="$PS1"
-	if [ $ISTERM_TESTING == "1" ]; then
-		__is_prior_prompt = "> "
+	if [[ $ISTERM_TESTING == "1" ]]; then
+		__is_prior_prompt="> "
 	fi
 	PS1="%{$(__is_prompt_start)%}$PS1%{$(__is_prompt_end)%}"
 }
diff --git a/shell/shellIntegration.fish b/shell/shellIntegration.fish
index d86163e..4420370 100644
--- a/shell/shellIntegration.fish
+++ b/shell/shellIntegration.fish
@@ -12,7 +12,7 @@ function __is_update_cwd --on-event fish_prompt; set __is_cwd (__is_escape_value
 
 __is_copy_function fish_prompt is_user_prompt
 
-if [ "$ISTERM_TESTING" == "1" ]
+if [ "$ISTERM_TESTING" = "1" ]
 	function is_user_prompt; printf '> '; end
 end