diff --git a/python/mscclpp/comm.py b/python/mscclpp/comm.py
index b4f11ae43..b3cc51f12 100644
--- a/python/mscclpp/comm.py
+++ b/python/mscclpp/comm.py
@@ -97,7 +97,7 @@ def make_connection(
             else:
                 endpoint = endpoints
             if endpoint.transport == Transport.Nvls:
-                connections[rank] = self.communicator.connct_nvls_collective(all_ranks, endpoint)
+                return self.communicator.connct_nvls_collective(all_ranks, endpoint)
             else:
                 connections[rank] = self.communicator.connect_on_setup(rank, 0, endpoint)
         self.communicator.setup()