diff --git a/tests/MailKitSimplified.Receiver.Tests/ImapReceiverUnitTests.cs b/tests/MailKitSimplified.Receiver.Tests/ImapReceiverUnitTests.cs
index 76970b4..e2c0b3c 100644
--- a/tests/MailKitSimplified.Receiver.Tests/ImapReceiverUnitTests.cs
+++ b/tests/MailKitSimplified.Receiver.Tests/ImapReceiverUnitTests.cs
@@ -115,7 +115,7 @@ public async Task DisposeAsync_WithImapReceiver()
         public async Task ConnectImapClientAsync_VerifyType()
         {
             // Act
-            var imapReceiver = await _imapReceiver.ConnectAuthenticatedImapClientAsync(It.IsAny<CancellationToken>());
+            var imapReceiver = await _imapReceiver.ConnectAuthenticatedImapClientAsync(It.IsAny<CancellationToken>(), It.IsAny<bool>());
             // Assert
             Assert.NotNull(imapReceiver);
             Assert.IsAssignableFrom<IImapClient>(imapReceiver);
diff --git a/tests/MailKitSimplified.Receiver.Tests/MailFolderMonitorUnitTests.cs b/tests/MailKitSimplified.Receiver.Tests/MailFolderMonitorUnitTests.cs
index 83e7dff..4ab26ab 100644
--- a/tests/MailKitSimplified.Receiver.Tests/MailFolderMonitorUnitTests.cs
+++ b/tests/MailKitSimplified.Receiver.Tests/MailFolderMonitorUnitTests.cs
@@ -30,7 +30,7 @@ public MailFolderMonitorUnitTests()
                 .Returns(_imapReceiverMock.Object).Verifiable();
             _imapReceiverMock.Setup(_ => _.ConnectMailFolderAsync(It.IsAny<CancellationToken>()))
                 .ReturnsAsync(_mailFolderMock.Object).Verifiable();
-            _imapReceiverMock.Setup(_ => _.ConnectAuthenticatedImapClientAsync(It.IsAny<CancellationToken>()))
+            _imapReceiverMock.Setup(_ => _.ConnectAuthenticatedImapClientAsync(It.IsAny<CancellationToken>(), It.IsAny<bool>()))
                 .ReturnsAsync(_imapClientMock.Object).Verifiable();
             _imapReceiverMock.SetupGet(_ => _.MailFolderClient)
                 .Returns(_mailFolderClientMock.Object).Verifiable();