diff --git a/SignalMetadataKitTests/src/SMKTestUtils.swift b/SignalMetadataKitTests/src/SMKTestUtils.swift
index 5d486ae..3a6cdac 100644
--- a/SignalMetadataKitTests/src/SMKTestUtils.swift
+++ b/SignalMetadataKitTests/src/SMKTestUtils.swift
@@ -54,7 +54,7 @@ class MockClient: NSObject {
         self.identityKeyPair = IdentityKeyPair.generate()
 
         let protocolStore = InMemorySignalProtocolStore(identity: identityKeyPair,
-                                                        deviceId: UInt32(bitPattern: deviceId))
+                                                        registrationId: UInt32(registrationId))
 
         sessionStore = protocolStore
         preKeyStore = protocolStore