diff --git a/d2graph/d2graph.go b/d2graph/d2graph.go
index d8a7ca3fb..49ff3e211 100644
--- a/d2graph/d2graph.go
+++ b/d2graph/d2graph.go
@@ -1446,7 +1446,7 @@ func (g *Graph) SortEdgesByAST() {
 		if len(e1.References) == 0 || len(e2.References) == 0 {
 			return i < j
 		}
-		return e1.References[0].Edge.Range.Before(edges[j].References[0].Edge.Range)
+		return e1.References[0].Edge.Range.Before(e2.References[0].Edge.Range)
 	})
 	g.Edges = edges
 }