diff --git a/src/app/dashboard/ai-coach/mistakes/page.tsx b/src/app/dashboard/ai-coach/mistakes/page.tsx
index d5aec99..f21bf8e 100644
--- a/src/app/dashboard/ai-coach/mistakes/page.tsx
+++ b/src/app/dashboard/ai-coach/mistakes/page.tsx
@@ -7,6 +7,7 @@ import { Card, CardContent } from "@/components/ui/card"
 import { Button } from "@/components/ui/button"
 import { BrainCircuit, RefreshCw, CheckCircle2, Flame, Sparkles, AlertTriangle, Target, Clock, TrendingDown } from "lucide-react"
 import { useActiveUser } from "@/hooks/useActiveUser"
+import { Badge } from "@/components/ui/badge"
 
 export default function MistakeTrackingPage() {
   const [mistakes, setMistakes] = useState<MistakePayload[]>([])