diff --git a/components/tree/__tests__/__snapshots__/demo.test.js.snap b/components/tree/__tests__/__snapshots__/demo.test.js.snap
index e779666adc..62a1cbe72c 100644
--- a/components/tree/__tests__/__snapshots__/demo.test.js.snap
+++ b/components/tree/__tests__/__snapshots__/demo.test.js.snap
@@ -1040,7 +1040,7 @@ exports[`renders ./components/tree/demo/search.md correctly 1`] = `
 <div>
   <span
     class="ant-input-search ant-input-affix-wrapper"
-    style="width:300px;"
+    style="margin-bottom:8px;"
   >
     <input
       class="ant-input"
diff --git a/components/tree/demo/search.md b/components/tree/demo/search.md
index 405960bcca..28f004f8ed 100644
--- a/components/tree/demo/search.md
+++ b/components/tree/demo/search.md
@@ -125,7 +125,7 @@ class SearchTree extends React.Component {
     });
     return (
       <div>
-        <Search style={{ width: 300 }} placeholder="Search" onChange={this.onChange} />
+        <Search style={{ marginBottom: 8 }} placeholder="Search" onChange={this.onChange} />
         <Tree
           onExpand={this.onExpand}
           expandedKeys={expandedKeys}