From 07785c9ae010ae7e04e340c5ee014b2bb18ff749 Mon Sep 17 00:00:00 2001
From: John Luo <jt.luo@mail.utoronto.ca>
Date: Thu, 7 Jan 2016 13:55:41 -0800
Subject: [PATCH] Reacting to new hosting API

---
 .../KestrelWebSocketHelpers.cs                              | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs b/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs
index 2b4edd2194..132da53e5b 100644
--- a/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs
+++ b/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs
@@ -47,11 +47,13 @@ namespace Microsoft.AspNet.WebSockets.Client.Test
 
             var application = new WebApplicationBuilder()
                 .UseConfiguration(config)
-                .UseServerFactory("Microsoft.AspNet.Server.Kestrel")
+                .UseServer("Microsoft.AspNet.Server.Kestrel")
                 .Configure(startup)
                 .Build();
 
-            return application.Start();
+            application.Start();
+
+            return application;
         }
     }
 }