From 75a15a0a93d1cf502ba48942bdf70b70be56df00 Mon Sep 17 00:00:00 2001
From: Olli Saarikivi <olsaarik@microsoft.com>
Date: Tue, 12 Sep 2023 18:48:58 +0000
Subject: [PATCH] Fix a test for API changes

---
 test/unit/core_tests.cc | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/test/unit/core_tests.cc b/test/unit/core_tests.cc
index e696fd508..f2bd70203 100644
--- a/test/unit/core_tests.cc
+++ b/test/unit/core_tests.cc
@@ -35,9 +35,8 @@ TEST_F(LocalCommunicatorTest, RegisterMemory) {
 TEST_F(LocalCommunicatorTest, SendMemoryToSelf) {
   int dummy[42];
   auto memory = comm->registerMemory(&dummy, sizeof(dummy), mscclpp::NoTransports);
-  comm->sendMemoryOnSetup(memory, 0, 0);
-  auto memoryFuture = comm->recvMemoryOnSetup(0, 0);
-  comm->setup();
+  comm->sendMemory(memory, 0, 0);
+  auto memoryFuture = comm->recvMemory(0, 0);
   auto sameMemory = memoryFuture.get();
   EXPECT_EQ(sameMemory.data(), memory.data());
   EXPECT_EQ(sameMemory.size(), memory.size());