diff --git a/cmd/hz/generator/router.go b/cmd/hz/generator/router.go
index 10f431657..db141b0e7 100644
--- a/cmd/hz/generator/router.go
+++ b/cmd/hz/generator/router.go
@@ -321,7 +321,10 @@ func (pkgGen *HttpPackageGenerator) updateRegister(pkg, rDir, pkgName string) er
 		}
 
 		insertReg := register.DepPkgAlias + ".Register(r)\n"
-		if bytes.Contains(file, []byte(insertReg)) {
+
+		registerPattern := `(?m)^\s*` + register.DepPkgAlias + `\s*\.\s*Register\(r\)\s*$`
+		re := regexp.MustCompile(registerPattern)
+		if re.Match(file) {
 			return fmt.Errorf("the router(%s) has been registered", insertReg)
 		}