diff --git a/src/utils/token-tracker.ts b/src/utils/token-tracker.ts
index bdc29fe..336593d 100644
--- a/src/utils/token-tracker.ts
+++ b/src/utils/token-tracker.ts
@@ -15,7 +15,7 @@ export class TokenTracker extends EventEmitter {
       const asyncLocalContext = process.asyncLocalContext as any;
       this.on('usage', () => {
         if (asyncLocalContext.available()) {
-          asyncLocalContext.ctx.chargeAmount = this.getTotalUsage();
+          asyncLocalContext.ctx.chargeAmount = this.getTotalUsage().totalTokens;
         }
       });