3 files changed, 7 insertions, 4 deletions

diff --git a/bin/task/context b/bin/task/context
index 06848d6..639f54a 100755
--- a/bin/task/context
+++ b/bin/task/context
@@ -16,6 +16,10 @@ PROC=$$
 
 context="$(cat ~/.task/context)"
 [[ "$*" =~ '-s' ]] && SET=true && shift
+[[ "$1" =~ ^[a-zA-Z] ]] && { 
+  echo "$1" > ~/.task/context
+  exit 0
+}
 time="$1"
 
 # The unit of time display. For example:
diff --git a/bin/task/shellrc b/bin/task/shellrc
index 7098ac0..6d7f109 100644
--- a/bin/task/shellrc
+++ b/bin/task/shellrc
@@ -1,7 +1,6 @@
 # task
 alias c='context'
 tk() { task rc.context=`cat ~/.task/context` $@; }
-tkc(){ echo $1 >~/.task/context; }
 alias tkcc='task context'
 tkcn() { echo none >~/.task/context; }
 tkcp() { echo plan >~/.task/context; }
diff --git a/zsh/_tkc.zsh b/zsh/_context.zsh
index 2e98d4f..dbb250d 100644
--- a/zsh/_tkc.zsh
+++ b/zsh/_context.zsh
@@ -1,7 +1,7 @@
-#compdef tkc
+#compdef context
 
-function _tkc() {
+function _context() {
   compadd -S '' $(sed -E '/^$/q' ~/log/plan.context.md | cut -d' ' -f1)
 }
 
-_tkc
+_context