diff --git a/src/chatbox.tsx b/src/chatbox.tsx
index 1c9298d..746139f 100644
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -169,7 +169,8 @@ export default function ChatBOX(props: {
     if (data.usage.prompt_tokens) {
       const userMessageToken = data.usage.prompt_tokens - aboveToken;
       console.log("set user message token");
-      chatStore.history.slice(-1)[0].token = userMessageToken;
+      chatStore.history.filter((msg) => !msg.hide).slice(-1)[0].token =
+        userMessageToken;
     }
 
     chatStore.history.push({