diff --git a/server/model/system/sys_auto_code.go b/server/model/system/sys_auto_code.go
index 47ba888cf250ca041ffd3b6070e4e4d45ec26a17..aac9cd2055bfaca92feb221ea155fd82d453828c 100644
--- a/server/model/system/sys_auto_code.go
+++ b/server/model/system/sys_auto_code.go
@@ -40,8 +40,8 @@ func (a *AutoCodeStruct) KeyWord() {
 // SuffixTest 处理_test 后缀
 // Author [SliverHorn](https://github.com/SliverHorn)
 func (a *AutoCodeStruct) SuffixTest() {
-	if strings.HasSuffix(a.Package, "test") {
-		a.Package = a.Package + "_"
+	if strings.HasSuffix(a.HumpPackageName, "test") {
+		a.HumpPackageName = a.HumpPackageName + "_"
 	}
 }