From db0ff660e581023c4219729ac9257794a1354d9f Mon Sep 17 00:00:00 2001
From: WBHarry <williambjrklund@gmail.com>
Date: Sun, 28 Dec 2025 19:25:19 +0100
Subject: [PATCH] .

---
 module/applications/dialogs/beastformDialog.mjs | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/module/applications/dialogs/beastformDialog.mjs b/module/applications/dialogs/beastformDialog.mjs
index c4a62a79..09a9222b 100644
--- a/module/applications/dialogs/beastformDialog.mjs
+++ b/module/applications/dialogs/beastformDialog.mjs
@@ -286,13 +286,9 @@ export default class BeastformDialog extends HandlebarsApplicationMixin(Applicat
                     if (data) {
                         if (!data.selectedImage) selected = null;
                         else {
-                            if (evolved) {
-                                if (evolved.usesDynamicToken) evolved.system.tokenRingImg = data.selectedImage;
-                                else evolved.system.tokenImg = data.selectedImage;
-                            } else {
-                                if (data.usesDynamicToken) selected.system.tokenRingImg = data.selectedImage;
-                                else selected.system.tokenImg = data.selectedImage;
-                            }
+                            const imageSource = evolved ?? selected;
+                            if (imageSource.usesDynamicToken) imageSource.system.tokenRingImg = data.selectedImage;
+                            else imageSource.system.tokenImg = data.selectedImage;
                         }
                     }