diff --git a/src/Nethermind/Nethermind.AuRa.Test/AuRaPluginTests.cs b/src/Nethermind/Nethermind.AuRa.Test/AuRaPluginTests.cs index 009035305dd..03c84f9b2b8 100644 --- a/src/Nethermind/Nethermind.AuRa.Test/AuRaPluginTests.cs +++ b/src/Nethermind/Nethermind.AuRa.Test/AuRaPluginTests.cs @@ -6,6 +6,7 @@ using Nethermind.Api; using Nethermind.Config; using Nethermind.Consensus.AuRa; +using Nethermind.Consensus.AuRa.InitializationSteps; using Nethermind.Logging; using Nethermind.Serialization.Json; using Nethermind.Specs.ChainSpecStyle; @@ -19,7 +20,7 @@ public class AuRaPluginTests public void Init_when_not_AuRa_doesnt_trow() { AuRaPlugin auRaPlugin = new(); - Action init = () => auRaPlugin.Init(new NethermindApi(new ConfigProvider(), new EthereumJsonSerializer(), new TestLogManager(), new ChainSpec())); + Action init = () => auRaPlugin.Init(new AuRaNethermindApi(new ConfigProvider(), new EthereumJsonSerializer(), new TestLogManager(), new ChainSpec())); init.Should().NotThrow(); } diff --git a/src/Nethermind/Nethermind.Consensus.AuRa/AuRaPlugin.cs b/src/Nethermind/Nethermind.Consensus.AuRa/AuRaPlugin.cs index 28987a17c93..adab94dde86 100644 --- a/src/Nethermind/Nethermind.Consensus.AuRa/AuRaPlugin.cs +++ b/src/Nethermind/Nethermind.Consensus.AuRa/AuRaPlugin.cs @@ -42,6 +42,7 @@ public ValueTask DisposeAsync() public Task Init(INethermindApi nethermindApi) { _nethermindApi = nethermindApi as AuRaNethermindApi; + _blockProducerStarter = new(_nethermindApi); return Task.CompletedTask; } @@ -69,7 +70,6 @@ public Task InitBlockProducer(ITxSource? additionalTxSource = nu { if (_nethermindApi is not null) { - _blockProducerStarter = new(_nethermindApi); return _blockProducerStarter!.BuildProducer(additionalTxSource); }