diff --git a/d2chaos/d2chaos.go b/d2chaos/d2chaos.go
index 2f9f670d5..7729454a8 100644
--- a/d2chaos/d2chaos.go
+++ b/d2chaos/d2chaos.go
@@ -246,10 +246,7 @@ func (gs *dslGenState) randStr(n int, inKey bool) string {
 func (gs *dslGenState) randShape() string {
 	for {
 		s := shapes[gs.rand.Intn(len(shapes))]
-		if s != "image" {
-			return s
-		}
-		if s != "sequence_diagram" {
+		if s != d2target.ShapeImage {
 			return s
 		}
 	}