diff --git a/tests/unit/light-volume.integration.test.tsx b/tests/unit/light-volume.integration.test.tsx
index 5ee816ea..5021e944 100644
--- a/tests/unit/light-volume.integration.test.tsx
+++ b/tests/unit/light-volume.integration.test.tsx
@@ -19,6 +19,7 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     updateSimulation: ReturnType<typeof vi.fn>;
     updateAssets: ReturnType<typeof vi.fn>;
     updateDocument: ReturnType<typeof vi.fn>;
+    setRenderEnabled: ReturnType<typeof vi.fn>;
     setViewMode: ReturnType<typeof vi.fn>;
     setDisplayMode: ReturnType<typeof vi.fn>;
     setCameraState: ReturnType<typeof vi.fn>;
@@ -46,6 +47,7 @@ const { MockViewportHost, viewportHostInstances } = vi.hoisted(() => {
     updateSimulation = vi.fn();
     updateAssets = vi.fn();
     updateDocument = vi.fn();
+    setRenderEnabled = vi.fn();
     setViewMode = vi.fn();
     setDisplayMode = vi.fn();
     setCameraState = vi.fn();