diff --git a/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs b/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs
index 4fd33d8eb6..d07e108c82 100644
--- a/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs
+++ b/test/Microsoft.AspNet.WebSockets.Client.Test/KestrelWebSocketHelpers.cs
@@ -46,7 +46,7 @@ namespace Microsoft.AspNet.WebSockets.Client.Test
             config["server.urls"] = "http://localhost:54321";
 
             var host = new WebHostBuilder(config)
-                .UseServer("Microsoft.AspNet.Server.Kestrel")
+                .UseServerFactory("Microsoft.AspNet.Server.Kestrel")
                 .UseStartup(startup)
                 .Build();