diff --git a/ui/litellm-dashboard/src/components/shared/advanced_date_picker.tsx b/ui/litellm-dashboard/src/components/shared/advanced_date_picker.tsx
index 8aac45b33e5..5d9605ce24a 100644
--- a/ui/litellm-dashboard/src/components/shared/advanced_date_picker.tsx
+++ b/ui/litellm-dashboard/src/components/shared/advanced_date_picker.tsx
@@ -1,6 +1,7 @@
 import { Calendar as CalendarIcon, Clock } from "lucide-react";
 // eslint-disable-next-line litellm-ui/no-banned-ui-imports
-import { Button, DateRangePickerValue } from "@tremor/react";
+import { DateRangePickerValue } from "@tremor/react";
+import { Button } from "@/components/ui/button";
 import moment from "moment";
 import React, { useCallback, useEffect, useRef, useState } from "react";