@@ -46,7 +46,10 @@ func untar() {

 		fatal(err)

 	}

 	// fully consume stdin in case it is zero padded

-	flush(os.Stdin)

+	if _, err := flush(os.Stdin); err != nil {

+		fatal(err)

+	}

+

 	os.Exit(0)

 }

@@ -65,8 +65,10 @@ func applyLayer() {

 		fatal(fmt.Errorf("unable to encode layerSize JSON: %s", err))

 	}

-	flush(os.Stdout)

-	flush(os.Stdin)

+	if _, err := flush(os.Stdin); err != nil {

+		fatal(err)

+	}

+

 	os.Exit(0)

 }

@@ -23,6 +23,6 @@ func fatal(err error) {

 // flush consumes all the bytes from the reader discarding

 // any errors

-func flush(r io.Reader) {

-	io.Copy(ioutil.Discard, r)

+func flush(r io.Reader) (bytes int64, err error) {

+	return io.Copy(ioutil.Discard, r)

 }