diff --git a/goutils/proto_build/tpls/main_gen.tpl b/goutils/proto_build/tpls/main_gen.tpl
index 4809b22..02f5bf4 100644
--- a/goutils/proto_build/tpls/main_gen.tpl
+++ b/goutils/proto_build/tpls/main_gen.tpl
@@ -2,8 +2,9 @@ package main
 
 import (
 	"{{.ProjectName}}/gen/go/service"
-	"{{.ProjectName}}/server/logics/info_logic"
-	"{{.ProjectName}}/server/logics/product_logic"
+	{{range .StructServiceNames}}
+	"{{$.ProjectName}}/server/logics/{{.LogicPackageName}}"
+	{{- end}}
 
 	"google.golang.org/grpc"
 )