diff --git a/components/float-button/__tests__/__snapshots__/demo-extend.test.ts.snap b/components/float-button/__tests__/__snapshots__/demo-extend.test.ts.snap
index 8e9c5799a2..993db3167a 100644
--- a/components/float-button/__tests__/__snapshots__/demo-extend.test.ts.snap
+++ b/components/float-button/__tests__/__snapshots__/demo-extend.test.ts.snap
@@ -855,6 +855,7 @@ Array [
     aria-checked="true"
     class="ant-switch ant-switch-checked"
     role="switch"
+    style="margin: 16px;"
     type="button"
   >
     <div
diff --git a/components/float-button/__tests__/__snapshots__/demo.test.ts.snap b/components/float-button/__tests__/__snapshots__/demo.test.ts.snap
index fbc7630841..b6efca230a 100644
--- a/components/float-button/__tests__/__snapshots__/demo.test.ts.snap
+++ b/components/float-button/__tests__/__snapshots__/demo.test.ts.snap
@@ -645,6 +645,7 @@ Array [
     aria-checked="true"
     class="ant-switch ant-switch-checked"
     role="switch"
+    style="margin:16px"
     type="button"
   >
     <div
diff --git a/components/float-button/demo/controlled.tsx b/components/float-button/demo/controlled.tsx
index d47f8b459e..7993acf7a9 100644
--- a/components/float-button/demo/controlled.tsx
+++ b/components/float-button/demo/controlled.tsx
@@ -20,7 +20,7 @@ const App: React.FC = () => {
         <FloatButton />
         <FloatButton icon={<CommentOutlined />} />
       </FloatButton.Group>
-      <Switch onChange={onChange} checked={open} />
+      <Switch onChange={onChange} checked={open} style={{ margin: 16 }} />
     </>
   );
 };