fix

ChainSafe · Jun 17, 2021 · d1b23b1 · d1b23b1
1 parent 62b7277
commit d1b23b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/babe/babe_test.go b/lib/babe/babe_test.go
@@ -92,7 +92,7 @@ func createTestService(t *testing.T, cfg *ServiceConfig) *Service {
 	}
 
 	cfg.BlockImportHandler = new(mocks.BlockImportHandler)
-	cfg.BlockImportHandler.(*mocks.BlockImportHandler).On("HandleBlockImport", mock.AnythingOfType("*types.Block"), mock.AnythingOfType("*storage.TrieState")).Return(nil)
+	cfg.BlockImportHandler.(*mocks.BlockImportHandler).On("HandleBlockProduced", mock.AnythingOfType("*types.Block"), mock.AnythingOfType("*storage.TrieState")).Return(nil)
 
 	if cfg.Keypair == nil {
 		cfg.Keypair, err = sr25519.GenerateKeypair()