diff --git a/src/_chatblade b/src/_chatblade
index c2bd4e6..62d12e6 100644
--- a/src/_chatblade
+++ b/src/_chatblade
@@ -15,7 +15,7 @@
 
 _chatblade() {
   local -a args
-  local context state_descr line
+  local context state state_descr line
   typeset -A opt_args
 
   args+=(
@@ -41,6 +41,8 @@ _chatblade() {
     '--session-rename[rename session]:new session:->sessions'
   )
 
+  _arguments -s $args
+
   case $state in
     (sessions)
       local -a sessions
@@ -48,8 +50,6 @@ _chatblade() {
       _describe 'session' sessions
       ;;
   esac
-
-  _arguments -s $args
 }
 
 _chatblade