Tweak in edge.rb

diff --git a/lib/graph.njae/edge.rb b/lib/graph.njae/edge.rb
index 6581033d2a885bbe4fe8beddf1eddfb38879b50a..f38a32b21e693bbf645c25bdcca159579f97dceb 100644 (file)
--- a/lib/graph.njae/edge.rb
+++ b/lib/graph.njae/edge.rb
@@ -31,7 +31,7 @@ module GraphNjae
     
     # Return the connection object that joins this Edge to the specified Vertex
     def connection_at(vertex)
-      self.connections.select {|c| c.end.equal?  vertex}.first
+      self.connections.find {|c| c.end.equal?  vertex}
     end
   end