diff --git a/src/core/Elsa.Core/StartupTasks/StartServiceBusTask.cs b/src/core/Elsa.Core/StartupTasks/StartServiceBusTask.cs index 0c7b7610e..af04e2c6b 100644 --- a/src/core/Elsa.Core/StartupTasks/StartServiceBusTask.cs +++ b/src/core/Elsa.Core/StartupTasks/StartServiceBusTask.cs @@ -1,15 +1,22 @@ +using System; using System.Threading; using System.Threading.Tasks; using Elsa.Messages; using Elsa.Services; -using Rebus.Bus; +using Rebus.ServiceProvider; namespace Elsa.StartupTasks { public class StartServiceBusTask : IStartupTask { - private readonly IBus _serviceBus; - public StartServiceBusTask(IBus serviceBus) => _serviceBus = serviceBus; - public Task ExecuteAsync(CancellationToken cancellationToken = default) => _serviceBus.Subscribe(); + private readonly IServiceProvider _serviceProvider; + public StartServiceBusTask(IServiceProvider serviceProvider) => _serviceProvider = serviceProvider; + + public Task ExecuteAsync(CancellationToken cancellationToken = default) + { + _serviceProvider.UseRebus(x => x.Subscribe()); + return Task.CompletedTask; + } } + } \ No newline at end of file