From 39573cd92261043fb16d78b88d305fe200dab3e5 Mon Sep 17 00:00:00 2001
From: mkorwel <matt.korwel@gmail.com>
Date: Sat, 18 Apr 2026 22:30:20 +0000
Subject: [PATCH] test: fix useModelCommand tests

---
 .../cli/src/ui/hooks/useModelCommand.test.tsx | 21 ++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/packages/cli/src/ui/hooks/useModelCommand.test.tsx b/packages/cli/src/ui/hooks/useModelCommand.test.tsx
index b93474e149..30513e5179 100644
--- a/packages/cli/src/ui/hooks/useModelCommand.test.tsx
+++ b/packages/cli/src/ui/hooks/useModelCommand.test.tsx
@@ -18,13 +18,23 @@ describe('useModelCommand', () => {
   }
 
   it('should initialize with the model dialog closed', async () => {
-    const { unmount } = await render(<TestComponent />);
+    const { unmount } = await render(
+      <TestComponent />,
+      undefined,
+      undefined,
+      true,
+    );
     expect(result.isModelDialogOpen).toBe(false);
     unmount();
   });
 
   it('should open the model dialog when openModelDialog is called', async () => {
-    const { unmount } = await render(<TestComponent />);
+    const { unmount } = await render(
+      <TestComponent />,
+      undefined,
+      undefined,
+      true,
+    );
 
     act(() => {
       result.openModelDialog();
@@ -35,7 +45,12 @@ describe('useModelCommand', () => {
   });
 
   it('should close the model dialog when closeModelDialog is called', async () => {
-    const { unmount } = await render(<TestComponent />);
+    const { unmount } = await render(
+      <TestComponent />,
+      undefined,
+      undefined,
+      true,
+    );
 
     // Open it first
     act(() => {