diff --git a/util/protogen/main.go b/util/protogen/main.go
index 41ed991..68c4da8 100644
--- a/util/protogen/main.go
+++ b/util/protogen/main.go
@@ -15,7 +15,7 @@ func main() {
 			//	continue
 			//}
 			imp := string(f.GoImportPath)
-			if strings.HasSuffix(imp, "/tree") { // || strings.HasSuffix(imp, "/control") {
+			if strings.HasSuffix(imp, "/tree") || strings.HasSuffix(imp, "/control") {
 				generateFile(gen, f)
 			}
 		}