diff --git a/.gitignore b/.gitignore
index c8ea253..d3448df 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,6 +5,8 @@
 .gradle/
 *.iml
 
+gradle.properties
+
 ##忽略输出文件夹
 out/
 build/
diff --git a/build.gradle b/build.gradle
index 28818c3..58b7b20 100644
--- a/build.gradle
+++ b/build.gradle
@@ -55,7 +55,7 @@ intellij {
     // 各种版本去这里找
     // https://www.jetbrains.com/intellij-repository/releases
     // 依赖的插件
-    plugins = ['DatabaseTools', 'Velocity']
+    plugins = ['java', 'DatabaseTools', 'Velocity']
     //Disables updating since-build attribute in plugin.xml
     updateSinceUntilBuild false
 }
diff --git a/src/main/java/com/sjhy/plugin/ui/SelectSavePath.java b/src/main/java/com/sjhy/plugin/ui/SelectSavePath.java
index 528ca0e..a4b63fa 100644
--- a/src/main/java/com/sjhy/plugin/ui/SelectSavePath.java
+++ b/src/main/java/com/sjhy/plugin/ui/SelectSavePath.java
@@ -1,6 +1,6 @@
 package com.sjhy.plugin.ui;
 
-//import com.intellij.ide.util.PackageChooserDialog;
+import com.intellij.ide.util.PackageChooserDialog;
 import com.intellij.openapi.fileChooser.FileChooser;
 import com.intellij.openapi.fileChooser.FileChooserDescriptorFactory;
 import com.intellij.openapi.module.Module;
@@ -9,8 +9,7 @@ import com.intellij.openapi.project.Project;
 import com.intellij.openapi.ui.Messages;
 import com.intellij.openapi.vfs.LocalFileSystem;
 import com.intellij.openapi.vfs.VirtualFile;
-//import com.intellij.psi.PsiPackage;
-import com.intellij.util.ExceptionUtil;
+import com.intellij.psi.PsiPackage;
 import com.sjhy.plugin.config.Settings;
 import com.sjhy.plugin.constants.MsgValue;
 import com.sjhy.plugin.constants.StrState;
@@ -19,15 +18,16 @@ import com.sjhy.plugin.entity.Template;
 import com.sjhy.plugin.entity.TemplateGroup;
 import com.sjhy.plugin.service.CodeGenerateService;
 import com.sjhy.plugin.service.TableInfoService;
-import com.sjhy.plugin.tool.*;
+import com.sjhy.plugin.tool.CacheDataUtils;
+import com.sjhy.plugin.tool.CurrGroupUtils;
+import com.sjhy.plugin.tool.ModuleUtils;
+import com.sjhy.plugin.tool.StringUtils;
 
 import javax.swing.*;
 import java.awt.*;
 import java.awt.event.*;
-import java.lang.reflect.Constructor;
-import java.lang.reflect.InvocationTargetException;
-import java.lang.reflect.Method;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.LinkedList;
 import java.util.List;
 
 /**
@@ -282,38 +282,14 @@ public class SelectSavePath extends JDialog {
 
         //添加包选择事件
         packageChooseButton.addActionListener(e -> {
-            // 这里不知道发生了什么，明明类是存在的但是编译时就是找不到，只用通过反射来使用
-            try {
-                // 构建dialog
-                Class<?> cls = Class.forName("com.intellij.ide.util.PackageChooserDialog");
-                Constructor<?> constructor = cls.getConstructor(String.class, Project.class);
-                Object dialog = constructor.newInstance("Package Chooser", project);
-                // 打开dialog窗口
-                Method show = dialog.getClass().getMethod("show");
-                show.invoke(dialog);
-                // 获取选中的包信息
-                Method getSelectedPackage = dialog.getClass().getMethod("getSelectedPackage");
-                Object psiPackage = getSelectedPackage.invoke(dialog);
-                // 获取名字
-                if (psiPackage != null) {
-                    Method getQualifiedName = psiPackage.getClass().getMethod("getQualifiedName");
-                    String packageName = (String) getQualifiedName.invoke(psiPackage);
-                    packageField.setText(packageName);
-                    // 刷新路径
-                    refreshPath();
-                }
-            } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException | InstantiationException | ClassNotFoundException e1) {
-                // 抛出异常信息
-                ExceptionUtil.rethrow(e1);
+            PackageChooserDialog dialog = new PackageChooserDialog("Package Chooser", project);
+            dialog.show();
+            PsiPackage psiPackage = dialog.getSelectedPackage();
+            if (psiPackage != null) {
+                packageField.setText(psiPackage.getQualifiedName());
+                // 刷新路径
+                refreshPath();
             }
-//            PackageChooserDialog dialog = new PackageChooserDialog("Package Chooser", project);
-//            dialog.show();
-//            PsiPackage psiPackage = dialog.getSelectedPackage();
-//            if (psiPackage != null) {
-//                packageField.setText(psiPackage.getQualifiedName());
-//                // 刷新路径
-//                refreshPath();
-//            }
         });
 
         // 添加包编辑框失去焦点事件