From 1a64fb81f3c121717d08673e95218630795d4ffa Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Fri, 3 Apr 2026 00:22:31 +0200
Subject: [PATCH] Refactor viewport-host.ts to improve conditional logic

---
 src/viewport-three/viewport-host.ts | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 10bd6032..8ad1a1d8 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -1824,13 +1824,14 @@ export class ViewportHost {
               entityKind: toolPreview.target.entityKind,
               audioAssetId: toolPreview.target.audioAssetId
             }
-          : {
-              kind: "model-instance",
-              assetId: toolPreview.target.assetId
-            }
-          : {
-              kind: "box-brush"
-            },
+          : toolPreview.target.kind === "model-instance"
+            ? {
+                kind: "model-instance",
+                assetId: toolPreview.target.assetId
+              }
+            : {
+                kind: "box-brush"
+              },
       center: toolPreview.center === null ? null : { ...toolPreview.center }
     };
 
@@ -1877,7 +1878,7 @@ export class ViewportHost {
     creationPreviewObject.visible = toolPreview.center !== null;
     this.scene.add(creationPreviewObject);
     this.creationPreviewObject = creationPreviewObject;
-    this.placementPreviewTargetKey = nextTargetKey;
+    this.creationPreviewTargetKey = nextTargetKey;
   }
 
   private render = () => {