diff --git a/style/components/checkbox.less b/style/components/checkbox.less
index eacd0a2b8c..adda7d8a78 100644
--- a/style/components/checkbox.less
+++ b/style/components/checkbox.less
@@ -1,4 +1,6 @@
-@checkboxWrapPrefixCls: ant-checkbox;
+.checkboxFn();
+.checkboxFn(@checkboxPrefixCls: ant-checkbox) {
+@checkboxWrapPrefixCls: @checkboxPrefixCls;
 @checkboxInnerPrefixCls: ~"@{checkboxWrapPrefixCls}-inner";
 
 /* 一般状态 */
@@ -149,3 +151,5 @@
 .@{checkboxWrapPrefixCls} + span {
   margin-left: 8px;
 }
+
+}
diff --git a/style/components/tree.less b/style/components/tree.less
index a99f610fe2..d845dc9bca 100644
--- a/style/components/tree.less
+++ b/style/components/tree.less
@@ -1,6 +1,5 @@
 @treePrefixCls: ant-tree;
-@import "./checkbox";
-@checkboxWrapPrefixCls: ant-tree-checkbox;
+.checkboxFn(@checkboxPrefixCls: ant-tree-checkbox);
 @import "../mixins/iconfont";
 @openIcon: '\e60f';
 .switcher_icon() {