diff --git a/components/badge/__tests__/__snapshots__/demo.test.js.snap b/components/badge/__tests__/__snapshots__/demo.test.js.snap
index 1f6601ee34..59524cc4e1 100644
--- a/components/badge/__tests__/__snapshots__/demo.test.js.snap
+++ b/components/badge/__tests__/__snapshots__/demo.test.js.snap
@@ -399,7 +399,8 @@ exports[`renders ./components/badge/demo/change.md correctly 1`] = `
       />
     </span>
     <span
-      class="ant-switch"
+      checked=""
+      class="ant-switch ant-switch-checked"
       tabindex="0"
     >
       <span
diff --git a/components/badge/demo/change.md b/components/badge/demo/change.md
index a56b044b30..ada549ab7e 100644
--- a/components/badge/demo/change.md
+++ b/components/badge/demo/change.md
@@ -60,7 +60,7 @@ class Demo extends React.Component {
           <Badge dot={this.state.show}>
             <a href="#" className="head-example" />
           </Badge>
-          <Switch onChange={this.onChange} />
+          <Switch onChange={this.onChange} checked={this.state.show} />
         </div>
       </div>
     );