From 945d8d9e567c3f36ae16cb674e22cf9d30e6eb11 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Sun, 18 Oct 2020 12:17:27 +0200
Subject: [PATCH] Fix service bus starter

---
 .../Elsa.Core/StartupTasks/StartServiceBusTask.cs | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

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<RunWorkflow>();
+        private readonly IServiceProvider _serviceProvider;
+        public StartServiceBusTask(IServiceProvider serviceProvider) => _serviceProvider = serviceProvider;
+        
+        public Task ExecuteAsync(CancellationToken cancellationToken = default)
+        {
+            _serviceProvider.UseRebus(x => x.Subscribe<RunWorkflow>());
+            return Task.CompletedTask;
+        }
     }
+    
 }
\ No newline at end of file