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 528d18f4..053974e0 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
@@ -3,6 +3,7 @@ package com.reajason.javaweb.memshell.tomcat.command;
 import com.reajason.javaweb.config.CommandConfig;
 import com.reajason.javaweb.config.ShellConfig;
 import com.reajason.javaweb.memshell.CommandGenerator;
+import com.reajason.javaweb.memshell.shelltool.command.CommandFilter;
 import com.reajason.javaweb.memshell.shelltool.command.CommandValve;
 import com.reajason.javaweb.util.ClassUtils;
 import org.junit.jupiter.params.ParameterizedTest;
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 6da11a8f..6146693a 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
@@ -3,6 +3,7 @@ package com.reajason.javaweb.memshell.tomcat.godzilla;
 import com.reajason.javaweb.config.GodzillaConfig;
 import com.reajason.javaweb.config.ShellConfig;
 import com.reajason.javaweb.memshell.GodzillaGenerator;
+import com.reajason.javaweb.memshell.shelltool.godzilla.GodzillaFilter;
 import com.reajason.javaweb.memshell.shelltool.godzilla.GodzillaValve;
 import com.reajason.javaweb.util.ClassUtils;
 import com.reajason.javaweb.util.CommonUtil;