diff --git a/template/common/babel.config.js b/template/common/babel.config.js
index a46c091..8ce24af 100644
--- a/template/common/babel.config.js
+++ b/template/common/babel.config.js
@@ -1,4 +1,8 @@
-const plugins = []
+const plugins = []
+
+if (process.env.UNI_OPT_TREESHAKINGNG) {
+  plugins.push(require('@dcloudio/vue-cli-plugin-uni-optimize/packages/babel-plugin-uni-api/index.js'))
+}
 
 if (process.env.UNI_PLATFORM === 'app-plus' && process.env.UNI_USING_V8) {
   const path = require('path')