From 78a28de2aac7d32388fe7b4c03e65d4b698f71c2 Mon Sep 17 00:00:00 2001
From: Jonathan Moody <103143855+moodyjon@users.noreply.github.com>
Date: Thu, 8 Sep 2022 08:52:20 -0500
Subject: [PATCH] Align style of generate() with generate_and_wait().

---
 lbry/testcase.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lbry/testcase.py b/lbry/testcase.py
index 75ece97ee..3f1095f49 100644
--- a/lbry/testcase.py
+++ b/lbry/testcase.py
@@ -327,9 +327,9 @@ class IntegrationTestCase(AsyncioTestCase):
     async def generate(self, blocks):
         """ Ask lbrycrd to generate some blocks and wait until ledger has them. """
         prepare = self.ledger.on_header.where(self.blockchain.is_expected_block)
-        height = self.blockchain.block_expected + blocks
         self.conductor.spv_node.server.synchronized.clear()
         await self.blockchain.generate(blocks)
+        height = self.blockchain.block_expected
         await prepare  # no guarantee that it didn't happen already, so start waiting from before calling generate
         while True:
             await self.conductor.spv_node.server.synchronized.wait()