diff --git a/tests/sync/connectblocks.go b/tests/sync/connectblocks.go
index 53a69754..4d89d079 100644
--- a/tests/sync/connectblocks.go
+++ b/tests/sync/connectblocks.go
@@ -32,11 +32,11 @@ func testConnectBlocks(t *testing.T, h *TestHandler) {
 				t.Fatal(err)
 			}
 
-			err = db.ConnectBlocks(sw, func(hash string, height uint32) {
-				if hash == upperHash {
-					close(ch)
-				}
-			}, true)
+				err = db.ConnectBlocks(sw, func(block *bchain.Block) {
+					if block != nil && block.Hash == upperHash {
+						close(ch)
+					}
+				}, true)
 			if err != nil && err != db.ErrOperationInterrupted {
 				t.Fatal(err)
 			}
diff --git a/tests/sync/handlefork.go b/tests/sync/handlefork.go
index 871d4fce..42253990 100644
--- a/tests/sync/handlefork.go
+++ b/tests/sync/handlefork.go
@@ -88,11 +88,11 @@ func testHandleFork(t *testing.T, h *TestHandler) {
 			chain.returnFakes = false
 
 			upperHash := fakeBlocks[len(fakeBlocks)-1].Hash
-			db.HandleFork(sw, rng.Upper, upperHash, func(hash string, height uint32) {
-				if hash == upperHash {
-					close(ch)
-				}
-			}, true)
+				db.HandleFork(sw, rng.Upper, upperHash, func(block *bchain.Block) {
+					if block != nil && block.Hash == upperHash {
+						close(ch)
+					}
+				}, true)
 
 			realBlocks := getRealBlocks(h, rng)
 			realTxs, err := getTxs(h, d, rng, realBlocks)