diff --git a/test/realtime/connection.test.js b/test/realtime/connection.test.js
index 5a6b7952c..c8551286c 100644
--- a/test/realtime/connection.test.js
+++ b/test/realtime/connection.test.js
@@ -335,7 +335,7 @@ define(['ably', 'shared_helper', 'async', 'chai'], function (Ably, Helper, async
           helper.closeAndFinish(done, realtime);
         });
         helper.recordPrivateApi('call.connectionManager.activeProtocol.getTransport');
-        helper.recordPrivateApi('replace.transport.onProtocolMessage');
+        helper.recordPrivateApi('call.transport.onProtocolMessage');
         helper.recordPrivateApi('call.protocolMessageFromDeserialized');
         connectionManager.activeProtocol.getTransport().onProtocolMessage(
           createPM({
diff --git a/test/realtime/presence.test.js b/test/realtime/presence.test.js
index 636c25954..f2de186fd 100644
--- a/test/realtime/presence.test.js
+++ b/test/realtime/presence.test.js
@@ -2052,6 +2052,7 @@ define(['ably', 'shared_helper', 'async', 'chai'], function (Ably, Helper, async
           },
           function (cb) {
             /* Inject a member locally */
+            helper.recordPrivateApi('call.channel.processMessage');
             channel
               .processMessage({
                 action: 14,