lib/aggro/node.rb in aggro-0.0.3 vs lib/aggro/node.rb in aggro-0.0.4

@@ -18,10 +18,12 @@
     def discover_publisher_endpoint
       message = Message::PublisherEndpointInquiry.new(Aggro.local_node.id)
       response = client.post(message)
 
       if response.is_a? Message::Endpoint
-        response.endpoint
+        port = URI.parse(response.endpoint).port
+
+        URI.parse(endpoint).tap { |uri| uri.port = port }.to_s
       else
         fail "Could not discover publisher endpoint for #{id}"
       end
     end
   end