diff --git a/src/main/java/burp/rule/utils/RuleTool.java b/src/main/java/burp/rule/utils/RuleTool.java
index d65ff87..1525d91 100644
--- a/src/main/java/burp/rule/utils/RuleTool.java
+++ b/src/main/java/burp/rule/utils/RuleTool.java
@@ -25,7 +25,7 @@ public class RuleTool {
             // 获取官方规则文件，在线更新写入
             FileOutputStream fileOutputStream = new FileOutputStream(this.rulesFilePath);
             fileOutputStream.write(httpResponse.body().bytes());
-            JOptionPane.showMessageDialog(null, "Config file updated successfully!", "Error",
+            JOptionPane.showMessageDialog(null, "Rules updated successfully!", "Info",
                     JOptionPane.INFORMATION_MESSAGE);
         } catch (Exception ignored) {
             JOptionPane.showMessageDialog(null, "Please check your network!", "Error",
diff --git a/src/main/java/burp/rule/utils/YamlTool.java b/src/main/java/burp/rule/utils/YamlTool.java
index ff8c7be..08de7be 100644
--- a/src/main/java/burp/rule/utils/YamlTool.java
+++ b/src/main/java/burp/rule/utils/YamlTool.java
@@ -1,6 +1,8 @@
 package burp.rule.utils;
 
 import java.util.Map;
+
+import burp.BurpExtender;
 import org.yaml.snakeyaml.DumperOptions;
 import org.yaml.snakeyaml.Yaml;
 import java.io.FileInputStream;