diff --git a/www/apps/ui/src/content/docs/components/icon-button.mdx b/www/apps/ui/src/content/docs/components/icon-button.mdx
index 54484fb271..98b5961d33 100644
--- a/www/apps/ui/src/content/docs/components/icon-button.mdx
+++ b/www/apps/ui/src/content/docs/components/icon-button.mdx
@@ -11,14 +11,14 @@ component: true
 ---
 
 ```tsx
-import { IconBadge } from "@medusajs/ui"
+import { IconButton } from "@medusajs/ui"
 import { Plus } from "@medusajs/icons"
 ```
 
 ```tsx
-<IconBadge>
+<IconButton>
   <Plus />
-</IconBadge>
+</IconButton>
 ```
 
 ## API Reference
diff --git a/www/apps/ui/src/props/button.tsx b/www/apps/ui/src/props/button.tsx
index 0e7fbd1a44..571456b9f3 100644
--- a/www/apps/ui/src/props/button.tsx
+++ b/www/apps/ui/src/props/button.tsx
@@ -14,7 +14,7 @@ const buttonProps: PropDataMap = [
     prop: "size",
     type: {
       type: "enum",
-      values: ["small", "base", "large", "xlarge"],
+      values: ["base", "large", "xlarge"],
     },
     defaultValue: "base",
   },