diff --git a/ui/litellm-dashboard/src/components/create_user_button.test.tsx b/ui/litellm-dashboard/src/components/create_user_button.test.tsx
new file mode 100644
index 00000000000..e40a1e0ac3c
--- /dev/null
+++ b/ui/litellm-dashboard/src/components/create_user_button.test.tsx
@@ -0,0 +1,35 @@
+import React from "react";
+import { render } from "@testing-library/react";
+import { describe, it, expect, vi } from "vitest";
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
+import Createuser from "./create_user_button";
+
+vi.mock("./networking", () => ({
+ userCreateCall: vi.fn(),
+ modelAvailableCall: vi.fn().mockResolvedValue({ data: [] }),
+ invitationCreateCall: vi.fn(),
+ getProxyUISettings: vi.fn().mockResolvedValue({
+ PROXY_BASE_URL: null,
+ PROXY_LOGOUT_URL: null,
+ DEFAULT_TEAM_DISABLED: false,
+ SSO_ENABLED: false,
+ }),
+ getProxyBaseUrl: vi.fn().mockReturnValue("http://localhost"),
+}));
+
+const createQueryClient = () =>
+ new QueryClient({
+ defaultOptions: { queries: { retry: false, gcTime: 0 } },
+ });
+
+describe("Create User Button", () => {
+ it("should render the create user button", () => {
+ const qc = createQueryClient();
+ const { getByText } = render(
+
+
+ ,
+ );
+ expect(getByText("Create User")).toBeInTheDocument();
+ });
+});
diff --git a/ui/litellm-dashboard/src/components/create_user_button.tsx b/ui/litellm-dashboard/src/components/create_user_button.tsx
index 9973c2e777c..6fb6f80c4b4 100644
--- a/ui/litellm-dashboard/src/components/create_user_button.tsx
+++ b/ui/litellm-dashboard/src/components/create_user_button.tsx
@@ -26,6 +26,7 @@ import { InfoCircleOutlined } from "@ant-design/icons";
import { getModelDisplayName } from "./key_team_helpers/fetch_available_models_team_key";
import { useQueryClient } from "@tanstack/react-query";
import NotificationsManager from "./molecules/notifications_manager";
+import TeamDropdown from "./common_components/team_dropdown";
// Helper function to generate UUID compatible across all environments
const generateUUID = (): string => {
@@ -201,19 +202,9 @@ const Createuser: React.FC = ({
))}
-
-
-
- {teams ? (
- teams.map((team: any) => (
-
- ))
- ) : (
-
- )}
-
+