s/save/sort duh

1 files changed, 4 insertions, 4 deletions

diff --git a/protoParse.go b/protoParse.go
index c8ac75c..e09bbc3 100644
--- a/protoParse.go
+++ b/protoParse.go
@@ -87,6 +87,10 @@ func (pf *File) protoParse() error {
 			curmsg.Unique = append(curmsg.Unique, newU)
 			msgvar.HasUnique = true
 		}
+		if strings.Contains(line, "`autogenpb:save") {
+			pf.DoSave = true
+		}
+
 		curmsg.Vars = append(curmsg.Vars, msgvar)
 	}
 	pf.makeSortTable()
@@ -205,10 +209,6 @@ func (pf *File) parseForMessage(line string) *MsgName {
 		msg.DoMarshal = true // http requires Marshal
 	}
 
-	if strings.Contains(line, "`autogenpb:autosave") {
-		pf.DoSave = true
-	}
-
 	if strings.Contains(line, "`autogenpb:gui") {
 		log.Info("got autogenpb:gui")
 		pf.DoGui = true