From 31c93175db70f4070bf2894be134fc1e2d54458f Mon Sep 17 00:00:00 2001
From: Geoffrey Goh <kg422@cornell.edu>
Date: Fri, 13 Nov 2015 00:20:14 -0800
Subject: [PATCH] error handling tweaks

---
 CodePushDownloadHandler.m | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/CodePushDownloadHandler.m b/CodePushDownloadHandler.m
index 5d1992a..024a3b8 100644
--- a/CodePushDownloadHandler.m
+++ b/CodePushDownloadHandler.m
@@ -60,17 +60,19 @@ failCallback:(void (^)(NSError *err))failCallback {
     
     if (bytesLeft) {
         self.failCallback([self.outputFileStream streamError]);
+        [self.outputFileStream close];
     }
 }
 
 - (void)connection:(NSURLConnection*)connection didFailWithError:(NSError*)error
 {
     self.failCallback(error);
+    [self.outputFileStream close];
 }
 
 -(void)connectionDidFinishLoading:(NSURLConnection *)connection {
-    [self.outputFileStream close];
     self.doneCallback();
+    [self.outputFileStream close];
 }
 
 @end
\ No newline at end of file