diff --git a/consensus/update_test.go b/consensus/update_test.go
index bc354b0a..e93daf6c 100644
--- a/consensus/update_test.go
+++ b/consensus/update_test.go
@@ -86,11 +86,8 @@ func TestApplyBlock(t *testing.T) {
 		prev := cs
 		cs, au = consensus.ApplyBlock(prev, b, bs, ancestorTimestamp(dbStore, b.ParentID, cs.AncestorDepth()))
 		dbStore.ApplyBlock(prev, au, true)
-		dbStore.AddCheckpoint(chain.Checkpoint{
-			Block:      b,
-			State:      cs,
-			Supplement: &bs,
-		})
+		dbStore.AddBlock(b, &bs)
+		dbStore.AddState(cs)
 		return
 	}
 	checkUpdateElements := func(au consensus.ApplyUpdate, addedSCEs, spentSCEs []types.SiacoinElement, addedSFEs, spentSFEs []types.SiafundElement) {
@@ -326,5 +323,4 @@ func TestApplyV2Block(t *testing.T) {
 		t.Error("missing siacoin element", 1)
 	}
 }
-
 */