From ea0ccf6a2394f74ee2f7356b63c8e8449987b82c Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Fri, 23 Oct 2015 12:42:57 +1100
Subject: [PATCH] Tidy

---
 cmd/torrent-pick/main.go | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/cmd/torrent-pick/main.go b/cmd/torrent-pick/main.go
index 3bea027e..928f11f4 100644
--- a/cmd/torrent-pick/main.go
+++ b/cmd/torrent-pick/main.go
@@ -159,19 +159,17 @@ func main() {
 		}
 
 		go func() {
+			defer close(done)
 			<-t.GotInfo()
-			files := t.Files()
-			for _, file := range files {
-				if file.Path() == rootGroup.Pick {
-
-					log.Printf("Downloading file: %s", file.Path())
-
-					srcReader := io.NewSectionReader(t.NewReader(), file.Offset(), file.Length())
-					io.Copy(dstWriter, srcReader)
-					close(done)
-					break
+			for _, file := range t.Files() {
+				if file.DisplayPath() != rootGroup.Pick {
+					continue
 				}
+				srcReader := io.NewSectionReader(t.NewReader(), file.Offset(), file.Length())
+				io.Copy(dstWriter, srcReader)
+				return
 			}
+			log.Print("file not found")
 		}()
 	}