diff --git a/tests/unit/transform-foundation.integration.test.tsx b/tests/unit/transform-foundation.integration.test.tsx
index 242ec3ee..b13a42c5 100644
--- a/tests/unit/transform-foundation.integration.test.tsx
+++ b/tests/unit/transform-foundation.integration.test.tsx
@@ -33,6 +33,7 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     setWhiteboxHoverLabelChangeHandler: ReturnType<typeof vi.fn>;
     setWhiteboxSelectionMode: ReturnType<typeof vi.fn>;
     setWhiteboxSnapSettings: ReturnType<typeof vi.fn>;
+    setGridVisible: ReturnType<typeof vi.fn>;
     setToolMode: ReturnType<typeof vi.fn>;
     setCreationPreview: ReturnType<typeof vi.fn>;
     setTransformSession: ReturnType<typeof vi.fn>;
@@ -62,6 +63,7 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     setWhiteboxHoverLabelChangeHandler = vi.fn();
     setWhiteboxSelectionMode = vi.fn();
     setWhiteboxSnapSettings = vi.fn();
+    setGridVisible = vi.fn();
     setToolMode = vi.fn();
     setCreationPreview = vi.fn();
     setTransformSession = vi.fn();