From 4fb40e5437e4b0b0a9ce1302f3884bf98ac517f4 Mon Sep 17 00:00:00 2001
From: Liamsi <Liamsi@users.noreply.github.com>
Date: Wed, 20 Jun 2018 20:04:43 -0700
Subject: [PATCH] cherry pick @melekes changes

---
 consensus/wal_generator.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/consensus/wal_generator.go b/consensus/wal_generator.go
index e45f94231..9a4b70595 100644
--- a/consensus/wal_generator.go
+++ b/consensus/wal_generator.go
@@ -92,10 +92,11 @@ func WALWithNBlocks(numBlocks int) (data []byte, err error) {
 
 	select {
 	case <-numBlocksWritten:
+		consensusState.Stop()
 		wr.Flush()
 		return b.Bytes(), nil
 	case <-time.After(1 * time.Minute):
-		wr.Flush()
+		consensusState.Stop()
 		return []byte{}, fmt.Errorf("waited too long for tendermint to produce %d blocks (grep logs for `wal_generator`)", numBlocks)
 	}
 }