Wait for I/O to finish before waiting for subprocesses to exit

1 files changed, 2 insertions, 3 deletions

diff --git a/xdelta3/go/src/regtest.go b/xdelta3/go/src/regtest.go
index 385c936..76d253c 100644
--- a/xdelta3/go/src/regtest.go
+++ b/xdelta3/go/src/regtest.go
@@ -52,7 +52,7 @@ func smokeTest(r *xdelta.Runner, p *xdelta.Program) {
         if decoded != target {
                 g.Panic(errors.New("It's not working!!!"))
         }
-        t.Wait(g)
+        t.Wait(g, enc, dec)
         fmt.Println("Smoketest pass")
 }
 
@@ -80,8 +80,7 @@ func offsetTest(r *xdelta.Runner, p *xdelta.Program, bufsize, offset, length int
 
         xdelta.WriteRstreams(t, "encode", seed, offset, length, enc.Srcin, enc.Stdin)
         xdelta.WriteRstreams(t, "decode", seed, offset, length, dec.Srcin, write)
-
-        t.Wait(g)
+        t.Wait(g, enc, dec)
 }
 
 func main() {