diff --git a/providers/tokenbucket/examples_test.go b/providers/tokenbucket/examples_test.go
index cb1db0760..eb5e4ffeb 100644
--- a/providers/tokenbucket/examples_test.go
+++ b/providers/tokenbucket/examples_test.go
@@ -1,7 +1,6 @@
 package tokenbucket
 
 import (
-	middleware "github.com/grpc-ecosystem/go-grpc-middleware/v2"
 	grpc_ratelimit "github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/ratelimit"
 	"github.com/juju/ratelimit"
 	"google.golang.org/grpc"
@@ -17,16 +16,15 @@ const (
 
 // Simple example of server initialization code.
 func Example() {
-
-	limiter := TockenBucketInterceptor{}
+	limiter := TokenBucketInterceptor{}
 	limiter.tokenBucket = ratelimit.NewBucket(rate, int64(tokenCapacity))
 
 	_ = grpc.NewServer(
-		middleware.WithUnaryServerChain(
-			grpc_ratelimit.UnaryServerInterceptor(limiter),
+		grpc.ChainUnaryInterceptor(
+			grpc_ratelimit.UnaryServerInterceptor(&limiter),
 		),
-		middleware.WithStreamServerChain(
-			grpc_ratelimit.UnaryServerInterceptor(limiter),
+		grpc.ChainStreamInterceptor(
+			grpc_ratelimit.StreamServerInterceptor(&limiter),
 		),
 	)
 }
diff --git a/providers/tokenbucket/limiter.go b/providers/tokenbucket/limiter.go
index 98ba5e511..7a464e787 100644
--- a/providers/tokenbucket/limiter.go
+++ b/providers/tokenbucket/limiter.go
@@ -14,7 +14,7 @@ type TokenBucketInterceptor struct {
 	tokenBucket *ratelimit.Bucket
 }
 
-// Limit Implement Limiter interface
+// Limit implements Limiter interface.
 func (r *TokenBucketInterceptor) Limit(_ context.Context) error {
 	// Take one token per request. This call doesn't block.
 	tokenRes := r.tokenBucket.TakeAvailable(1)