Improve zsh completion

2 files changed, 2 insertions, 1 deletions

diff --git a/alias b/alias
index f72d343..34c5e76 100644
--- a/alias
+++ b/alias
@@ -37,6 +37,7 @@ alias vu='vim -u NONE'
 alias vq='vim ~/buffer'
 alias cdv='cd ~/.vim/vim-init'
 vs() { which $1 && vim $(which $1); }
+[ "$shell" = zsh ] && compdef vs=which
 
 # ssh
 alias cdssh='cd ~/.ssh'
diff --git a/bin/task/shellrc b/bin/task/shellrc
index 6f2e5d3..a7025d0 100644
--- a/bin/task/shellrc
+++ b/bin/task/shellrc
@@ -3,7 +3,7 @@ tk() { task rc.context=`cat ~/.task/context` $@; }
 tkc(){ echo $1 >~/.task/context; }
 tkcn() { echo none >~/.task/context; }
 tkch() { echo home >~/.task/context; }
-[ $shell = zsh ] && compdef tk=task
+[ "$shell" = zsh ] && compdef tk=task
 alias tka='tk annotate'
 alias tkd='tk done'
 alias tkD='tk delete'