diff --git a/SignalMetadataKitTests/src/SMKTestUtils.swift b/SignalMetadataKitTests/src/SMKTestUtils.swift
index d608aee..eae3167 100644
--- a/SignalMetadataKitTests/src/SMKTestUtils.swift
+++ b/SignalMetadataKitTests/src/SMKTestUtils.swift
@@ -93,7 +93,7 @@ class MockClient: NSObject {
         let identityKeyPair = self.identityStore.identityKeyPair(nil)!
         let signature = try! Ed25519.sign((keyPair.publicKey as NSData).prependKeyType() as Data, with: identityKeyPair)
         let signedPreKey = SignedPreKeyRecord(id: signedPreKeyId, keyPair: keyPair, signature: signature, generatedAt: generatedAt)
-        self.signedPreKeyStore.storeSignedPreKey(signedPreKeyId, signedPreKeyRecord: signedPreKey)
+        self.signedPreKeyStore.storeSignedPreKey(signedPreKeyId, signedPreKeyRecord: signedPreKey, protocolContext: nil)
         return signedPreKey
     }