From e5e34362fe9972c931a2bdfe68ce2f29d17b1d1f Mon Sep 17 00:00:00 2001
From: Bernard Xie <bernard@terrastruct.com>
Date: Wed, 16 Nov 2022 18:44:43 -0800
Subject: [PATCH] PR Comments

---
 cmd/d2/main.go | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/cmd/d2/main.go b/cmd/d2/main.go
index 277b61008..6331db81c 100644
--- a/cmd/d2/main.go
+++ b/cmd/d2/main.go
@@ -120,14 +120,14 @@ func run(ctx context.Context, ms *xmain.State) (err error) {
 		if err != nil {
 			return err
 		}
+		defer func() error {
+			err = png.Cleanup(pw, browser)
+			if err != nil {
+				return err
+			}
+			return nil
+		}()
 	}
-	defer func() error {
-		err = png.Cleanup(pw, browser)
-		if err != nil {
-			return err
-		}
-		return nil
-	}()
 
 	if *watchFlag {
 		if inputPath == "-" {