diff --git a/tests/unit/transform-foundation.integration.test.tsx b/tests/unit/transform-foundation.integration.test.tsx
index d81bb745..66ece9b7 100644
--- a/tests/unit/transform-foundation.integration.test.tsx
+++ b/tests/unit/transform-foundation.integration.test.tsx
@@ -30,6 +30,8 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     setTransformSessionChangeHandler: ReturnType<typeof vi.fn>;
     setTransformCommitHandler: ReturnType<typeof vi.fn>;
     setTransformCancelHandler: ReturnType<typeof vi.fn>;
+    setWhiteboxHoverLabelChangeHandler: ReturnType<typeof vi.fn>;
+    setWhiteboxSelectionMode: ReturnType<typeof vi.fn>;
     setWhiteboxSnapSettings: ReturnType<typeof vi.fn>;
     setToolMode: ReturnType<typeof vi.fn>;
     setCreationPreview: ReturnType<typeof vi.fn>;
@@ -57,6 +59,8 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     setTransformSessionChangeHandler = vi.fn();
     setTransformCommitHandler = vi.fn();
     setTransformCancelHandler = vi.fn();
+    setWhiteboxHoverLabelChangeHandler = vi.fn();
+    setWhiteboxSelectionMode = vi.fn();
     setWhiteboxSnapSettings = vi.fn();
     setToolMode = vi.fn();
     setCreationPreview = vi.fn();
diff --git a/tests/unit/viewport-canvas.test.tsx b/tests/unit/viewport-canvas.test.tsx
index 5dc06e77..12b459cd 100644
--- a/tests/unit/viewport-canvas.test.tsx
+++ b/tests/unit/viewport-canvas.test.tsx
@@ -24,6 +24,8 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     setTransformSessionChangeHandler: ReturnType<typeof vi.fn>;
     setTransformCommitHandler: ReturnType<typeof vi.fn>;
     setTransformCancelHandler: ReturnType<typeof vi.fn>;
+    setWhiteboxHoverLabelChangeHandler: ReturnType<typeof vi.fn>;
+    setWhiteboxSelectionMode: ReturnType<typeof vi.fn>;
     setWhiteboxSnapSettings: ReturnType<typeof vi.fn>;
     setToolMode: ReturnType<typeof vi.fn>;
     setCreationPreview: ReturnType<typeof vi.fn>;
@@ -48,6 +50,8 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     setTransformSessionChangeHandler = vi.fn();
     setTransformCommitHandler = vi.fn();
     setTransformCancelHandler = vi.fn();
+    setWhiteboxHoverLabelChangeHandler = vi.fn();
+    setWhiteboxSelectionMode = vi.fn();
     setWhiteboxSnapSettings = vi.fn();
     setToolMode = vi.fn();
     setCreationPreview = vi.fn();
@@ -107,6 +111,7 @@ describe("ViewportCanvas", () => {
         projectAssets={sceneDocument.assets}
         loadedModelAssets={{}}
         loadedImageAssets={{}}
+        whiteboxSelectionMode="object"
         whiteboxSnapEnabled
         whiteboxSnapStep={1}
         selection={{ kind: "none" }}
@@ -156,6 +161,7 @@ describe("ViewportCanvas", () => {
         projectAssets={sceneDocument.assets}
         loadedModelAssets={{}}
         loadedImageAssets={{}}
+        whiteboxSelectionMode="object"
         whiteboxSnapEnabled
         whiteboxSnapStep={1}
         selection={{ kind: "none" }}