diff --git a/processor/blockprocessor/initialize_test.go b/processor/blockprocessor/initialize_test.go
index f93fbf36c..1c72ffea2 100644
--- a/processor/blockprocessor/initialize_test.go
+++ b/processor/blockprocessor/initialize_test.go
@@ -64,9 +64,9 @@ func TestRunMigration(t *testing.T) {
 		AlgodToken:     "AAAAA",
 	}
 
-	ctx := context.Background()
+	ctx, cancel := context.WithCancel(context.Background())
 	// migrate 3 rounds
-	err = InitializeLedgerSimple(ctx, logrus.New(), 3, &opts)
+	err = InitializeLedgerSimple(ctx, cancel, logrus.New(), 3, &opts)
 	assert.NoError(t, err)
 	log, _ := test2.NewNullLogger()
 	l, err := util.MakeLedger(log, false, &genesis, opts.IndexerDatadir)
@@ -76,7 +76,8 @@ func TestRunMigration(t *testing.T) {
 	l.Close()
 
 	// migration continues from last round
-	err = InitializeLedgerSimple(ctx, logrus.New(), 6, &opts)
+	ctx, cancel = context.WithCancel(context.Background())
+	err = InitializeLedgerSimple(ctx, cancel, logrus.New(), 6, &opts)
 	assert.NoError(t, err)
 
 	l, err = util.MakeLedger(log, false, &genesis, opts.IndexerDatadir)