diff --git a/src/components/ui/badge/index.tsx b/src/components/ui/badge/index.tsx
index 924716776c..a2edc9c06e 100644
--- a/src/components/ui/badge/index.tsx
+++ b/src/components/ui/badge/index.tsx
@@ -122,7 +122,7 @@ function Badge({
 }: { className?: string } & IBadgeProps) {
   return (
     <ContextView
-      className={badgeStyle({ action, variant, class: className })}
+      className={badgeStyle({ action, variant, className })}
       {...props}
       context={{
         action,