diff --git a/transport/rateLimit.ts b/transport/rateLimit.ts
index 2e6901eb..17a2d96b 100644
--- a/transport/rateLimit.ts
+++ b/transport/rateLimit.ts
@@ -127,6 +127,11 @@ export class LeakyBucketRateLimit {
     this.intervalHandle = undefined;
   }
 
+  resetBudget() {
+    this.stopLeak();
+    this.budgetConsumed = 0;
+  }
+
   close() {
     this.stopLeak();
   }