fix in augmented remove block

Chia-Network · Dec 2, 2024 · 7deca59 · 7deca59
1 parent 401a2d5
commit 7deca59
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/chia/util/augmented_chain.py b/chia/util/augmented_chain.py
@@ -53,10 +53,9 @@ def add_extra_block(self, block: FullBlock, block_record: BlockRecord) -> None:
 
     def remove_extra_block(self, hh: bytes32) -> None:
         if hh in self._extra_blocks:
+            assert self._underlying.contains_block(hh)
             block_record = self._extra_blocks.pop(hh)[1]
-            assert block_record.height == self._peak_height
             del self._height_to_hash[block_record.height]
-            self._peak_height = uint32(self._peak_height - 1)
 
     # BlocksProtocol
     async def lookup_block_generators(self, header_hash: bytes32, generator_refs: set[uint32]) -> dict[uint32, bytes]: