From 2187bf78a86a13d341844d4bb18cee26813a1e48 Mon Sep 17 00:00:00 2001
From: Alexander Wang <alex@terrastruct.com>
Date: Wed, 20 Mar 2024 16:21:40 -0700
Subject: [PATCH] fix

---
 d2chaos/d2chaos.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/d2chaos/d2chaos.go b/d2chaos/d2chaos.go
index bf6961d4e..42445a974 100644
--- a/d2chaos/d2chaos.go
+++ b/d2chaos/d2chaos.go
@@ -137,9 +137,6 @@ func (gs *dslGenState) edge() error {
 		if err != nil {
 			return err
 		}
-		if src == dst && gs.nodeShapes[dst] == d2target.ShapeSequenceDiagram {
-			break
-		}
 		if gs.findOuterSequenceDiagram(src) == gs.findOuterSequenceDiagram(dst) {
 			break
 		}
@@ -149,6 +146,10 @@ func (gs *dslGenState) edge() error {
 		}
 	}
 
+	if src == dst && gs.nodeShapes[dst] == d2target.ShapeSequenceDiagram {
+		return nil
+	}
+
 	srcArrow := "-"
 	if gs.randBool() {
 		srcArrow = "<"