From e4d2a4441447bafb0ee552b80bf1fd039cb5c849 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Wed, 15 Apr 2026 07:57:00 +0200
Subject: [PATCH] Update transform-session.ts with specific geometry types

---
 src/core/transform-session.ts | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/core/transform-session.ts b/src/core/transform-session.ts
index 6f062d0a..1ad8a621 100644
--- a/src/core/transform-session.ts
+++ b/src/core/transform-session.ts
@@ -10,6 +10,9 @@ import {
   type Brush,
   type BrushGeometry,
   type BrushKind,
+  type BoxBrushGeometry,
+  type RadialPrismBrushGeometry,
+  type WedgeBrushGeometry,
   type WhiteboxEdgeId,
   type WhiteboxFaceId,
   type WhiteboxVertexId
@@ -259,7 +262,7 @@ function createBrushSnapshotFromTarget(
         center: target.initialCenter,
         rotationDegrees: target.initialRotationDegrees,
         size: target.initialSize,
-        geometry: target.initialGeometry as Parameters<typeof createBoxBrush>[0]["geometry"]
+        geometry: target.initialGeometry as BoxBrushGeometry
       });
     case "wedge":
       return createWedgeBrush({
@@ -267,7 +270,7 @@ function createBrushSnapshotFromTarget(
         center: target.initialCenter,
         rotationDegrees: target.initialRotationDegrees,
         size: target.initialSize,
-        geometry: target.initialGeometry as Parameters<typeof createWedgeBrush>[0]["geometry"]
+        geometry: target.initialGeometry as WedgeBrushGeometry
       });
     case "radialPrism":
       return createRadialPrismBrush({
@@ -276,10 +279,7 @@ function createBrushSnapshotFromTarget(
         rotationDegrees: target.initialRotationDegrees,
         size: target.initialSize,
         sideCount: target.sideCount,
-        geometry:
-          target.initialGeometry as Parameters<
-            typeof createRadialPrismBrush
-          >[0]["geometry"]
+        geometry: target.initialGeometry as RadialPrismBrushGeometry
       });
   }
 }