diff --git a/components/tree/__tests__/__snapshots__/demo.test.js.snap b/components/tree/__tests__/__snapshots__/demo.test.js.snap
index 01ce91b3d9..b3abe7b7dd 100644
--- a/components/tree/__tests__/__snapshots__/demo.test.js.snap
+++ b/components/tree/__tests__/__snapshots__/demo.test.js.snap
@@ -367,7 +367,7 @@ exports[`test renders ./components/tree/demo/basic-controlled.md correctly 1`] =
 
 exports[`test renders ./components/tree/demo/draggable.md correctly 1`] = `
 <ul
-  class="ant-tree"
+  class="draggable-tree ant-tree"
   role="tree-node"
   unselectable="true">
   <li