correctly identify the two primary structs

1 files changed, 19 insertions, 2 deletions

diff --git a/addMutex.go b/addMutex.go
index ff2aec5..cfd31b3 100644
--- a/addMutex.go
+++ b/addMutex.go
@@ -69,8 +69,25 @@ func (pb *Files) addMutex(f *File) error {
 
 // is this struct supposed to have a Mutex added?
 func (pf *File) structMatch(line string) bool {
-	for _, msg := range pf.MsgNames {
-		start := "type " + msg.Name + " struct {"
+	var msg *MsgName
+	var start string
+
+	msg = pf.Bases
+	start = "type " + msg.Name + " struct {"
+	if strings.HasPrefix(line, start) {
+		msg.MutexFound = true
+		return true
+	}
+
+	msg = pf.Base
+	start = "type " + msg.Name + " struct {"
+	if strings.HasPrefix(line, start) {
+		msg.MutexFound = true
+		return true
+	}
+
+	for _, msg = range pf.MsgNames {
+		start = "type " + msg.Name + " struct {"
 		if strings.HasPrefix(line, start) {
 			msg.MutexFound = true
 			return true