diff --git a/src/main.js b/src/main.js
index b1dd91a8c13c4f6e4d490912963658e052c475fe..f967c68de517612c8b0e912082b15503ca964612 100644
--- a/src/main.js
+++ b/src/main.js
@@ -19,7 +19,7 @@ import './permission' // permission control
 import './utils/error-log' // error log
 import request from '@/utils/request'
 import * as filters from './filters' // global filters
-import VueHighlightJS from 'vue-highlight.js'
+import VueHighlightJS from 'vue-highlightjs'
 
 /**
  * If you don't want to use mock-server
diff --git a/src/views/system/generate/index.vue b/src/views/system/generate/index.vue
index 93357cceb5e2fe6637f96e1cfcac2d843ffccbb4..54fe492f529dec697e8c4e1441a2fc4d7c4c8672 100644
--- a/src/views/system/generate/index.vue
+++ b/src/views/system/generate/index.vue
@@ -124,9 +124,11 @@
     <el-drawer
       :visible.sync="drawerVisible"
       direction="rtl">
-      <highlight-code lang="java">
-        {{ preview_content }}
-      </highlight-code>
+      <pre v-highlightjs>
+        <code class="php">
+          {{ preview_content }}
+        </code>
+      </pre>
     </el-drawer>
     <el-dialog title="创建模块" :visible="moduleCreateVisible" :destroy-on-close="true" width="40%" @close="moduleCreateVisible = false">
       <create-module @close="closeCreateModule" />