diff --git a/gcpkms/keysource_test.go b/gcpkms/keysource_test.go
index 2904eb34b..a37fb5b6e 100644
--- a/gcpkms/keysource_test.go
+++ b/gcpkms/keysource_test.go
@@ -7,8 +7,8 @@ import (
 	"testing"
 	"time"
 
+	"cloud.google.com/go/kms/apiv1/kmspb"
 	"github.com/stretchr/testify/assert"
-	kmspb "google.golang.org/genproto/googleapis/cloud/kms/v1"
 	"google.golang.org/grpc"
 )
 
diff --git a/gcpkms/mock_kms_server_test.go b/gcpkms/mock_kms_server_test.go
index 91b474747..bd2f30d60 100644
--- a/gcpkms/mock_kms_server_test.go
+++ b/gcpkms/mock_kms_server_test.go
@@ -23,12 +23,11 @@ import (
 	"io"
 	"strings"
 
-	kmspb "google.golang.org/genproto/googleapis/cloud/kms/v1"
-	"google.golang.org/protobuf/proto"
-	"google.golang.org/protobuf/types/known/anypb"
-
+	"cloud.google.com/go/kms/apiv1/kmspb"
 	"google.golang.org/genproto/googleapis/rpc/status"
 	"google.golang.org/grpc/metadata"
+	"google.golang.org/protobuf/proto"
+	"google.golang.org/protobuf/types/known/anypb"
 )
 
 var _ = io.EOF