diff --git a/src/commands/auth/login.sh b/src/commands/auth/login.sh
index 3434a9a..aeef7ea 100644
--- a/src/commands/auth/login.sh
+++ b/src/commands/auth/login.sh
@@ -64,6 +64,7 @@ wx-login(){
           wxi-stop
         fi
 
+        TOKEN=$VAULT_LOGIN
         wxi-config login
         ;;
       token)
diff --git a/wx b/wx
index e12b9b3..7a9a445 100755
--- a/wx
+++ b/wx
@@ -448,6 +448,7 @@ wx-login(){
           wxi-stop
         fi
 
+        TOKEN=$VAULT_LOGIN
         wxi-config login
         ;;
       token)