diff --git a/generator/src/test/java/com/reajason/javaweb/memshell/packer/JspPackerTest.java b/generator/src/test/java/com/reajason/javaweb/memshell/packer/JspPackerTest.java
index 8916e84a..e4bec708 100644
--- a/generator/src/test/java/com/reajason/javaweb/memshell/packer/JspPackerTest.java
+++ b/generator/src/test/java/com/reajason/javaweb/memshell/packer/JspPackerTest.java
@@ -17,8 +17,7 @@ class JspPackerTest {
         GenerateResult generateResult = GenerateResult.builder()
                 .injectorClassName("hehe")
                 .injectorBytesBase64Str("hehe").build();
-        String jspContent = new String(new JspPacker().pack(generateResult));
-        assertTrue(jspContent.contains("String className = \"hehe\";"));
+        String jspContent = new JspPacker().pack(generateResult);
         assertTrue(jspContent.contains("String base64Str = \"hehe\";"));
     }
 }
\ No newline at end of file
diff --git a/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/command/CommandFilterTest.java b/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/command/CommandFilterTest.java
index 76d976a5..675ebf1a 100644
--- a/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/command/CommandFilterTest.java
+++ b/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/command/CommandFilterTest.java
@@ -39,6 +39,7 @@ class CommandFilterTest {
     @MethodSource("casesProvider")
     void generate(String shellType, Class<?> clazz, String className) {
         ShellConfig generateConfig = new ShellConfig();
+        generateConfig.setServer(Server.Tomcat);
         CommandConfig commandConfig = CommandConfig.builder()
                 .shellClass(clazz)
                 .shellClassName(className)
diff --git a/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/godzilla/GodzillaTest.java b/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/godzilla/GodzillaTest.java
index e1d4da76..e89b0de3 100644
--- a/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/godzilla/GodzillaTest.java
+++ b/generator/src/test/java/com/reajason/javaweb/memshell/tomcat/godzilla/GodzillaTest.java
@@ -1,5 +1,6 @@
 package com.reajason.javaweb.memshell.tomcat.godzilla;
 
+import com.reajason.javaweb.memshell.Server;
 import com.reajason.javaweb.memshell.ShellType;
 import com.reajason.javaweb.memshell.config.GodzillaConfig;
 import com.reajason.javaweb.memshell.config.ShellConfig;
@@ -23,8 +24,7 @@ import static org.junit.jupiter.params.provider.Arguments.arguments;
  * @since 2024/11/24
  */
 class GodzillaTest {
-    ShellConfig config = new ShellConfig();
-
+    ShellConfig config = ShellConfig.builder().server(Server.Tomcat).build();
     GodzillaConfig.GodzillaConfigBuilder<?, ?> shellConfigBuilder = GodzillaConfig.builder()
             .pass(CommonUtil.getRandomString(6))
             .key(CommonUtil.getRandomString(6))