diff --git a/src/main/java/com/google/crypto/tink/tinkey/TinkeyTestKmsClient.java b/src/main/java/com/google/crypto/tink/tinkey/TinkeyTestKmsClient.java
index 53d4d99..4a77e4e 100644
--- a/src/main/java/com/google/crypto/tink/tinkey/TinkeyTestKmsClient.java
+++ b/src/main/java/com/google/crypto/tink/tinkey/TinkeyTestKmsClient.java
@@ -21,6 +21,7 @@
 import com.google.crypto.tink.InsecureSecretKeyAccess;
 import com.google.crypto.tink.KeysetHandle;
 import com.google.crypto.tink.KmsClient;
+import com.google.crypto.tink.RegistryConfiguration;
 import com.google.crypto.tink.TinkJsonProtoKeysetFormat;
 import java.io.IOException;
 import java.nio.file.Files;
@@ -104,6 +105,6 @@ public Aead getAead(String keyUri) throws GeneralSecurityException {
     checkCredentials();
     String keyset = stripPrefix(prefix, keyUri);
     return TinkJsonProtoKeysetFormat.parseKeyset(keyset, InsecureSecretKeyAccess.get())
-        .getPrimitive(Aead.class);
+        .getPrimitive(RegistryConfiguration.get(), Aead.class);
   }
 }