diff --git a/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift b/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift
index 370ddae08..c5eb784c0 100644
--- a/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift	
+++ b/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift	
@@ -139,7 +139,7 @@ extension MessageReceiver {
         ).inserted(db)
     }
     
-    public static func updateDisappearingMessagesConfigurationIfNeeded(
+    internal static func updateDisappearingMessagesConfigurationIfNeeded(
         _ db: Database,
         threadId: String,
         threadVariant: SessionThread.Variant,
diff --git a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
index fa2bd65f8..391c78f25 100644
--- a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
+++ b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
@@ -108,12 +108,13 @@ class MessageReceiverDisappearingMessagesSpec: QuickSpec {
                 it("updates the local config properly") {
                     mockStorage.write { db in
                         do {
-                            try MessageReceiver.updateDisappearingMessagesConfigurationIfNeeded(
+                            try MessageReceiver.handle(
                                 db,
                                 threadId: "TestId",
                                 threadVariant: .contact,
                                 message: mockMessage,
-                                proto: mockProto
+                                serverExpirationTimestamp: nil,
+                                associatedWithProto: mockProto
                             )
                         }
                         catch {