diff --git a/ui/litellm-dashboard/src/app/(dashboard)/teams/components/modals/CreateTeamModal.tsx b/ui/litellm-dashboard/src/app/(dashboard)/teams/components/modals/CreateTeamModal.tsx
index f82f34850d7..3ec6a7ee0b6 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/teams/components/modals/CreateTeamModal.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/teams/components/modals/CreateTeamModal.tsx
@@ -12,6 +12,7 @@ import MCPServerSelector from "@/components/mcp_server_management/MCPServerSelec
 import AgentSelector from "@/components/agent_management/AgentSelector";
 import PremiumLoggingSettings from "@/components/common_components/PremiumLoggingSettings";
 import ModelAliasManager from "@/components/common_components/ModelAliasManager";
+import { ModelBudgetEditor } from "@/components/OldTeams";
 import React, { useEffect, useState } from "react";
 import { useQueryClient } from "@tanstack/react-query";
 import NotificationsManager from "@/components/molecules/notifications_manager";
@@ -494,7 +495,7 @@ const CreateTeamModal = ({
             }
             name="team_member_model_max_budget"
           >
-            <TextInput placeholder={'{"gpt-3.5-turbo": {"max_budget": 10.0}}'} />
+            <ModelBudgetEditor availableModels={modelsToPick} />
           </Form.Item>
 
           <Form.Item label="Max Budget (USD)" name="max_budget">
diff --git a/ui/litellm-dashboard/src/components/OldTeams.tsx b/ui/litellm-dashboard/src/components/OldTeams.tsx
index 103467d622f..d4ed9d029ab 100644
--- a/ui/litellm-dashboard/src/components/OldTeams.tsx
+++ b/ui/litellm-dashboard/src/components/OldTeams.tsx
@@ -188,7 +188,7 @@ interface ModelBudgetEditorProps {
   availableModels: string[];
 }
 
-const ModelBudgetEditor: React.FC<ModelBudgetEditorProps> = ({ value, onChange, availableModels }) => {
+export const ModelBudgetEditor: React.FC<ModelBudgetEditorProps> = ({ value, onChange, availableModels }) => {
   const [rows, setRows] = useState<ModelBudgetRow[]>(() => {
     if (!value || Object.keys(value).length === 0) return [];
     return Object.entries(value).map(([model, cfg]) => ({