diff --git a/test/amqp.cpp b/test/amqp.cpp
index 8fafe3b..4e64259 100644
--- a/test/amqp.cpp
+++ b/test/amqp.cpp
@@ -129,7 +129,7 @@ TEST(AmqpTest, Connection_loginPlain_failed)
 
     StrictAmqpMock mock;
     InSequence seq;
-    EXPECT_CALL(mock, amqp_login_plain(AmqpMock::defaultConnectionState, StrEq("vhost"), 100, 200, 0, AMQP_SASL_METHOD_PLAIN, StrEq("username"), StrEq("password")))
+    EXPECT_CALL(mock, amqp_login_plain(AmqpMock::defaultConnectionState, StrEq("vhost"), 200, 131072, 100, AMQP_SASL_METHOD_PLAIN, StrEq("username"), StrEq("password")))
         .WillOnce(Return(amqp_rpc_reply_t {
             static_cast<amqp_response_type_enum>(AMQP_RESPONSE_LIBRARY_EXCEPTION),
             { 0, nullptr },
@@ -147,7 +147,7 @@ TEST(AmqpTest, Connection_loginPlain_success)
 
     StrictAmqpMock mock;
     InSequence seq;
-    EXPECT_CALL(mock, amqp_login_plain(AmqpMock::defaultConnectionState, StrEq("vhost"), 100, 200, 0, AMQP_SASL_METHOD_PLAIN, StrEq("username"), StrEq("password")))
+    EXPECT_CALL(mock, amqp_login_plain(AmqpMock::defaultConnectionState, StrEq("vhost"), 200, 131072, 100, AMQP_SASL_METHOD_PLAIN, StrEq("username"), StrEq("password")))
         .WillOnce(Return(AmqpMock::defaultRpcReply));
 
     EXPECT_NO_THROW(con.login_plain("username", "password", "vhost", 100, 200));