diff --git a/client/src/components/__tests__/ToolsTab.test.tsx b/client/src/components/__tests__/ToolsTab.test.tsx
index 2a45065..f75c8f9 100644
--- a/client/src/components/__tests__/ToolsTab.test.tsx
+++ b/client/src/components/__tests__/ToolsTab.test.tsx
@@ -16,6 +16,16 @@ describe("ToolsTab", () => {
         },
       },
     },
+    {
+      name: "tool3",
+      description: "Integer tool",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          count: { type: "integer" as const },
+        },
+      },
+    },
     {
       name: "tool2",
       description: "Second tool",
@@ -69,4 +79,28 @@ describe("ToolsTab", () => {
     const newInput = screen.getByRole("spinbutton") as HTMLInputElement;
     expect(newInput.value).toBe("");
   });
+
+  it("should handle integer type inputs", () => {
+    renderToolsTab({
+      selectedTool: mockTools[2],
+    });
+
+    // Verify input is rendered as a number input
+    const input = screen.getByRole("spinbutton") as HTMLInputElement;
+    expect(input.type).toBe("text"); // This will fail - should be "number"
+
+    // Enter an integer value
+    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: /submit/i });
+    fireEvent.click(submitButton);
+    
+    expect(defaultProps.callTool).toHaveBeenCalledWith(
+      mockTools[2].name,
+      { count: 42 }, // Should be number 42, not string "42"
+      expect.any(Function)
+    );
+  });
 });