diff --git a/aead/kms_envelope_aead_key_manager_test.go b/aead/kms_envelope_aead_key_manager_test.go
index 1164016..848afc8 100644
--- a/aead/kms_envelope_aead_key_manager_test.go
+++ b/aead/kms_envelope_aead_key_manager_test.go
@@ -21,6 +21,7 @@ import (
 
 	"google.golang.org/protobuf/proto"
 	"github.com/tink-crypto/tink-go/v2/aead"
+	"github.com/tink-crypto/tink-go/v2/core/registry"
 	"github.com/tink-crypto/tink-go/v2/keyset"
 	"github.com/tink-crypto/tink-go/v2/mac"
 	"github.com/tink-crypto/tink-go/v2/testing/fakekms"
@@ -64,6 +65,12 @@ func TestNewKMSEnvelopeAEADKeyWithInvalidSerializedKeyFormat(t *testing.T) {
 	if err != nil {
 		t.Fatalf("fakekms.NewKeyURI() err = %v", err)
 	}
+	fakeClient, err := fakekms.NewClient(keyURI)
+	if err != nil {
+		t.Fatalf("fakekms.NewClient() err = %v", err)
+	}
+	registry.RegisterKMSClient(fakeClient)
+	defer registry.ClearKMSClients()
 
 	// Create DEK template with unset embedded key parameters.
 	dekFormat := &ctrhmacpb.AesCtrHmacAeadKeyFormat{