diff --git a/src/button/__tests__/index.test.tsx b/src/button/__tests__/index.test.tsx
index 6fb4df277..7e1378a2c 100644
--- a/src/button/__tests__/index.test.tsx
+++ b/src/button/__tests__/index.test.tsx
@@ -33,6 +33,11 @@ describe('Button', () => {
expect(container.firstChild).toHaveClass('custom-class');
});
+ it('applies size className', () => {
+ const { container } = render();
+ expect(container.firstChild).toHaveClass('ant-btn-sm');
+ });
+
it('passes other props to AntdButton', () => {
const { getByText } = render();
expect(getByText('Primary').parentNode).toHaveClass('ant-btn-primary');
diff --git a/src/button/index.tsx b/src/button/index.tsx
index 9fb9dc386..b18ac868f 100644
--- a/src/button/index.tsx
+++ b/src/button/index.tsx
@@ -14,7 +14,7 @@ export default function Button({
...rest
}: ButtonProps) {
return (
-
+
{icon && {icon}}
{children && (
{children}