diff --git a/util/protogen/main.go b/util/protogen/main.go
index 85c6ded..1a287ba 100644
--- a/util/protogen/main.go
+++ b/util/protogen/main.go
@@ -16,11 +16,7 @@ var (
 func main() {
 	protogen.Options{}.Run(func(gen *protogen.Plugin) error {
 		for _, f := range gen.Files {
-			//if !f.Generate {
-			//	continue
-			//}
-			imp := string(f.GoImportPath)
-			if strings.HasSuffix(imp, "/tree") || strings.HasSuffix(imp, "/control") {
+			if f.Generate {
 				generateFile(gen, f)
 			}
 		}