diff --git a/scripts/site/generateColorLess.js b/scripts/site/generateColorLess.js
index 3250630f..51fcc211 100644
--- a/scripts/site/generateColorLess.js
+++ b/scripts/site/generateColorLess.js
@@ -3,6 +3,7 @@ const path = require('path');
 const { generateTheme } = require('antd-theme-generator');
 
 const options = {
+  antDir: path.join(__dirname, '../node_modules/antd'),
   stylesDir: path.join(__dirname, '../../site/src/'),
   antdStylesDir: path.join(__dirname, '../../components'),
   varFile: path.join(__dirname, '../../components/style/themes/default.less'),