From cd0f0154c18e1be3e2bcddb0f86cea3f6d048421 Mon Sep 17 00:00:00 2001
From: axeleron007 <andmis@gmail.com>
Date: Tue, 28 Sep 2021 11:11:01 +0300
Subject: [PATCH] Some fixes after pull request (merge with master)

---
 .../elsa-workflow-test-panel/elsa-workflow-test-panel.tsx     | 4 ++--
 src/samples/server/Elsa.Samples.Server.Host/Startup.cs        | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/designer/elsa-workflows-studio/src/components/screens/workflow-definition-editor/elsa-workflow-test-panel/elsa-workflow-test-panel.tsx b/src/designer/elsa-workflows-studio/src/components/screens/workflow-definition-editor/elsa-workflow-test-panel/elsa-workflow-test-panel.tsx
index a586e9323..acb4578de 100644
--- a/src/designer/elsa-workflows-studio/src/components/screens/workflow-definition-editor/elsa-workflow-test-panel/elsa-workflow-test-panel.tsx
+++ b/src/designer/elsa-workflows-studio/src/components/screens/workflow-definition-editor/elsa-workflow-test-panel/elsa-workflow-test-panel.tsx
@@ -65,7 +65,7 @@ export class ElsaWorkflowTestPanel {
     this.message = null;
     this.workflowTestActivityMessages = [];
     eventBus.emit(EventTypes.TestActivityMessageReceived, this, null);
-    const elsaClient = this.createClient();
+    const client = await createElsaClient(this.serverUrl);
 
     const request: WorkflowTestExecuteRequest = {
       workflowDefinitionId: this.workflowDefinition.definitionId,
@@ -73,7 +73,7 @@ export class ElsaWorkflowTestPanel {
       signalRConnectionId: this.signalRConnectionId
     };
 
-    await elsaClient.workflowTestApi.execute(request);
+    await client.workflowTestApi.execute(request);
   }
 
   render() {
diff --git a/src/samples/server/Elsa.Samples.Server.Host/Startup.cs b/src/samples/server/Elsa.Samples.Server.Host/Startup.cs
index 6c53f0875..ec9f522ab 100644
--- a/src/samples/server/Elsa.Samples.Server.Host/Startup.cs
+++ b/src/samples/server/Elsa.Samples.Server.Host/Startup.cs
@@ -1,5 +1,6 @@
 using System.Collections.Generic;
 using Elsa.Retention.Extensions;
+using Elsa.Server.Api.Hubs;
 using Elsa.Server.Hangfire.Extensions;
 using Hangfire;
 using Microsoft.AspNetCore.Builder;