diff --git a/project/allProject.js b/project/allProject.js
index d5810d91..747f297d 100644
--- a/project/allProject.js
+++ b/project/allProject.js
@@ -124,7 +124,7 @@ const generateMain = project => {
         const install = function (Vue) {
           if (install.installed) return Promise.resolve()
           else{
-            let contexts = require.context('.', true, /(\\/.+)[\\\/]App[^\\\/]+\\.vue$/)
+            let contexts = require.context('.', true, /(\\/.+)[\\\/]${project=='dvui'?'Ai':'App'}[^\\\/]+\\.vue$/)
             if (contexts) {
               contexts.keys().map(e => {
                 if (contexts(e).default) {
diff --git a/project/build.js b/project/build.js
index 7670dfe5..06f862ce 100644
--- a/project/build.js
+++ b/project/build.js
@@ -124,7 +124,7 @@ const start = () => {
         const install = function (Vue) {
           if (install.installed) return Promise.resolve()
           else{
-            let contexts = require.context('.', true, /(\\/.+)\\/App[^\\/]+\\.vue$/)
+            let contexts = require.context('.', true, /(\\/.+)\\/${prj == 'dvui' ? 'Ai' : 'App'}[^\\/]+\\.vue$/)
             if (contexts) {
               contexts.keys().map(e => {
                 if (contexts(e).default) {