diff --git a/packages/ui/src/components/shadcn/ui/badge.tsx b/packages/ui/src/components/shadcn/ui/badge.tsx
index b5276b0bdee..e20c873c222 100644
--- a/packages/ui/src/components/shadcn/ui/badge.tsx
+++ b/packages/ui/src/components/shadcn/ui/badge.tsx
@@ -9,8 +9,8 @@ const badgeVariants = cva(
     variants: {
       variant: {
         default: 'bg-surface-75 text-foreground-light border border-strong',
-        warning: 'bg-warning/10 text-warning border border-border-warning',
-        success: 'bg-brand/10 text-brand border border-border-brand',
+        warning: 'bg-warning/10 text-warning-600 border border-warning-500',
+        success: 'bg-brand/10 text-brand-600 border border-brand-500',
         destructive: 'bg-destructive/10 text-destructive border border-border-destructive',
         // Secondary is invisible
         secondary: