diff --git a/waku/common/rate_limit/token_bucket.nim b/waku/common/rate_limit/token_bucket.nim
index bb4baad99..9dcb2e7d1 100644
--- a/waku/common/rate_limit/token_bucket.nim
+++ b/waku/common/rate_limit/token_bucket.nim
@@ -109,10 +109,11 @@ proc update(bucket: TokenBucket, currentTime: Moment) =
   else:
     updateStrict(bucket, currentTime)
 
-## Returns the available capacity of the bucket: (budget, budgetCap)
 proc getAvailableCapacity*(
     bucket: TokenBucket, currentTime: Moment = Moment.now()
 ): tuple[budget: int, budgetCap: int] =
+  ## Returns the available capacity of the bucket: (budget, budgetCap)
+
   if periodElapsed(bucket, currentTime):
     case bucket.replenishMode
     of ReplenishMode.Strict: