1 files changed, 5 insertions, 4 deletions
diff --git a/zsh/.zshrc b/zsh/.zshrc
index ab6638c..79494e0 100644
--- a/zsh/.zshrc
+++ b/zsh/.zshrc
@@ -58,10 +58,6 @@ function preexec {
     print -Pn "\e];$cmd:q\a"
 }
 
-if [[ $TERM == xterm-termite && $DISPLAY == ":0" ]]; then
-    . /etc/profile.d/vte.sh
-    __vte_osc7
-fi
 
 # Load all configs
 #for f in ~/.config/zsh/zsh.d/*.zsh
@@ -120,3 +116,8 @@ if (( ${+terminfo[smkx]} )) && (( ${+terminfo[rmkx]} )); then
     zle -N zle-line-init
     zle -N zle-line-finish
 fi
+
+if [[ -n "$VTE_VERSION" ]]; then
+    source /etc/profile.d/vte.sh
+    __vte_prompt_command
+fi