diff --git a/client/src/components/__tests__/ToolsTab.test.tsx b/client/src/components/__tests__/ToolsTab.test.tsx
index b36085e..624a72e 100644
--- a/client/src/components/__tests__/ToolsTab.test.tsx
+++ b/client/src/components/__tests__/ToolsTab.test.tsx
@@ -85,15 +85,11 @@ describe("ToolsTab", () => {
       selectedTool: mockTools[1], // Use the tool with integer type
     });
 
-    // Verify input is rendered as a number input
-    const input = screen.getByRole("spinbutton") as HTMLInputElement;
-    expect(input.type).toBe("number"); // Integer type should be treated as number
-
-    // Enter an integer value
+    const input = screen.getByRole("textbox", { name: /count/i }) as HTMLInputElement;
+    expect(input).toHaveProperty("type", "number");
     fireEvent.change(input, { target: { value: "42" } });
     expect(input.value).toBe("42");
 
-    // Verify the callTool function receives the value as a number
     const submitButton = screen.getByRole("button", { name: /run tool/i });
     fireEvent.click(submitButton);