diff --git a/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift b/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift
index c5eb784c0..370ddae08 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)
     }
     
-    internal static func updateDisappearingMessagesConfigurationIfNeeded(
+    public static func updateDisappearingMessagesConfigurationIfNeeded(
         _ db: Database,
         threadId: String,
         threadVariant: SessionThread.Variant,
diff --git a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
index f0996cb0a..0404f8dfa 100644
--- a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
+++ b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
@@ -107,13 +107,12 @@ class MessageReceiverDisappearingMessagesSpec: QuickSpec {
                 // MARK: -- updates the local config properly
                 it("updates the local config properly") {
                     mockStorage.write { db in
-                        try MessageReceiver.handle(
+                        try MessageReceiver.updateDisappearingMessagesConfigurationIfNeeded(
                             db,
                             threadId: "TestId",
                             threadVariant: .contact,
                             message: mockMessage,
-                            serverExpirationTimestamp: nil,
-                            associatedWithProto: mockProto
+                            proto: mockProto
                         )
                     }