clean up

ezekg · Nov 21, 2015 · a1a2ecb · a1a2ecb
1 parent 5d0a058
commit a1a2ecb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -38,7 +38,7 @@ func main() {
 
 		sniff := func(fileName string, hunk *diff.Hunk) {
 			errs <- func() error {
-				defer func() { wg.Done() }()
+				defer wg.Done()
 				return hound.Sniff(fileName, hunk)
 			}()
 		}
@@ -47,8 +47,9 @@ func main() {
 			fileName := fileDiff.NewName
 			hunks := fileDiff.GetHunks()
 
+			wg.Add(len(hunks))
+
 			for _, hunk := range hunks {
-				wg.Add(1)
 				go sniff(fileName, hunk)
 			}
 		}