From 5d8aa1ee2a75b23dc08f6e00f92114bd604b2d13 Mon Sep 17 00:00:00 2001
From: BrennanConroy <brecon@microsoft.com>
Date: Tue, 14 Nov 2017 14:44:15 -0800
Subject: [PATCH] Fix a couple flaky tests (#1116)

---
 src/Microsoft.AspNetCore.SignalR.Client.Core/HubConnection.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.AspNetCore.SignalR.Client.Core/HubConnection.cs b/src/Microsoft.AspNetCore.SignalR.Client.Core/HubConnection.cs
index 9caa61a92b..7faaf21488 100644
--- a/src/Microsoft.AspNetCore.SignalR.Client.Core/HubConnection.cs
+++ b/src/Microsoft.AspNetCore.SignalR.Client.Core/HubConnection.cs
@@ -126,6 +126,7 @@ namespace Microsoft.AspNetCore.SignalR.Client
         private async Task DisposeAsyncCore()
         {
             await _connection.DisposeAsync();
+            await Closed;
         }
 
         // TODO: Client return values/tasks?
@@ -334,7 +335,7 @@ namespace Microsoft.AspNetCore.SignalR.Client
             }
         }
 
-        private Task Shutdown(Exception ex = null)
+        private void Shutdown(Exception ex = null)
         {
             _logger.ShutdownConnection();
             if (ex != null)
@@ -360,7 +361,6 @@ namespace Microsoft.AspNetCore.SignalR.Client
                 }
                 _pendingCalls.Clear();
             }
-            return Task.CompletedTask;
         }
 
         private async Task DispatchInvocationAsync(InvocationMessage invocation, CancellationToken cancellationToken)