diff --git a/discord/ext/commands/cooldowns.py b/discord/ext/commands/cooldowns.py index 7aadf0506..38b2fab9e 100644 --- a/discord/ext/commands/cooldowns.py +++ b/discord/ext/commands/cooldowns.py @@ -253,9 +253,9 @@ class CooldownMapping: return bucket - def update_rate_limit(self, message: Message, current: Optional[float] = None) -> Optional[float]: + def update_rate_limit(self, message: Message, current: Optional[float] = None, tokens: int = 1) -> Optional[float]: bucket = self.get_bucket(message, current) - return bucket.update_rate_limit(current) + return bucket.update_rate_limit(current, tokens=tokens) class DynamicCooldownMapping(CooldownMapping):