diff --git a/zsh.d/prompts/pure/pure.zsh b/zsh.d/prompts/pure/pure.zsh
index 5f7b274..b3a1094 100644
--- a/zsh.d/prompts/pure/pure.zsh
+++ b/zsh.d/prompts/pure/pure.zsh
@@ -749,11 +749,13 @@ prompt_pure_setup() {
     add-zle-hook-widget zle-keymap-select prompt_pure_update_vim_prompt_widget
   fi
 
-  # If a virtualenv is activated, display it in grey.
-  PROMPT='%(12V.%F{$prompt_pure_colors[virtualenv]}%12v%f .)'
+  # If a virtualenv is activated, display it
+  PROMPT='%(12V.%F{$prompt_pure_colors[virtualenv]}%12v%f.)'
 
-  # Add iterm2 mark (if enabled in iterm)
-  PROMPT+=%{$(iterm2_prompt_mark)%}
+  # Add iterm2 mark (if iterm integration is on)
+  if typeset -f iterm2_prompt_mark > /dev/null; then
+    PROMPT+=%{$(iterm2_prompt_mark)%}
+  fi
   # Prompt turns red if the previous command didn't exit with 0.
   local prompt_indicator='%(?.%F{$prompt_pure_colors[prompt:success]}.%F{$prompt_pure_colors[prompt:error]})${prompt_pure_state[prompt]}%f '
   PROMPT+=$prompt_indicator