From 2ef28f61ffa5142495d308ffff246faef26eab65 Mon Sep 17 00:00:00 2001
From: kortschak <dan.kortschak@adelaide.edu.au>
Date: Wed, 20 May 2015 16:43:44 +0930
Subject: [PATCH] concrete: don't return a different Edge type than was added

Updates #53.
---
 concrete/mutablegr.go | 2 +-
 concrete/mutdir.go    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/concrete/mutablegr.go b/concrete/mutablegr.go
index 45eb3ed6..1a367468 100644
--- a/concrete/mutablegr.go
+++ b/concrete/mutablegr.go
@@ -170,7 +170,7 @@ func (g *Graph) EdgeBetween(n, neigh graph.Node) graph.Edge {
 		return nil
 	}
 
-	return g.neighbors[n.ID()][neigh.ID()]
+	return g.neighbors[n.ID()][neigh.ID()].Edge
 }
 
 func (g *Graph) NodeExists(n graph.Node) bool {
diff --git a/concrete/mutdir.go b/concrete/mutdir.go
index 65ff05c3..f5db9351 100644
--- a/concrete/mutdir.go
+++ b/concrete/mutdir.go
@@ -151,7 +151,7 @@ func (g *DirectedGraph) EdgeTo(n, succ graph.Node) graph.Edge {
 	if !ok {
 		return nil
 	}
-	return edge
+	return edge.Edge
 }
 
 func (g *DirectedGraph) Predecessors(n graph.Node) []graph.Node {