From 68417cf380f84406f73a795bde947e4cba284f53 Mon Sep 17 00:00:00 2001
From: fulcanelly <dallasxfulcanelly@gmail.com>
Date: Wed, 9 Oct 2024 23:49:14 +0300
Subject: [PATCH] fix: state switching

---
 package.json       | 2 +-
 src/state/state.ts | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/package.json b/package.json
index 7ef6ac1..be5fd49 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "chat-toolkit",
-  "version": "0.0.5",
+  "version": "0.1.5",
   "description": "",
   "author": "",
   "private": false,
diff --git a/src/state/state.ts b/src/state/state.ts
index 11e198a..e633cbb 100644
--- a/src/state/state.ts
+++ b/src/state/state.ts
@@ -401,6 +401,7 @@ async function handleStateSwitch(e: SwitchStateError, context: Context, setState
     await context.manage.state.delete()
 
     await context.manage.state.save(e.stateToSwitch, e.state_arguments)
+    setStateData(context.allStates[e.stateToSwitch] as any, e.state_arguments)
   } catch (e) {
     console.error(e)
     throw e