diff --git a/consensus/wal_generator.go b/consensus/wal_generator.go
index 9a4b70595..e45f94231 100644
--- a/consensus/wal_generator.go
+++ b/consensus/wal_generator.go
@@ -92,11 +92,10 @@ 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):
-		consensusState.Stop()
+		wr.Flush()
 		return []byte{}, fmt.Errorf("waited too long for tendermint to produce %d blocks (grep logs for `wal_generator`)", numBlocks)
 	}
 }