diff --git a/bip-0158/gentestvectors.go b/bip-0158/gentestvectors.go
index bf6d9a7..686217f 100644
--- a/bip-0158/gentestvectors.go
+++ b/bip-0158/gentestvectors.go
@@ -290,12 +290,10 @@ func main() {
 				fmt.Println("Error writing test case to output: ", err.Error())
 				return
 			}
+
+			testBlockIndex++
 		}
 
 		prevBasicHeader = basicHeader
-
-		if uint32(height) == testBlockHeights[testBlockIndex].height {
-			testBlockIndex++
-		}
 	}
 }