diff --git a/src/Openchain.Server/Startup.cs b/src/Openchain.Server/Startup.cs
index d3195e8..e768b38 100644
--- a/src/Openchain.Server/Startup.cs
+++ b/src/Openchain.Server/Startup.cs
@@ -116,7 +116,7 @@ public void Configure(IApplicationBuilder app, IHostingEnvironment env, ILoggerF
             {
                 if (bool.Parse(configuration["enable_transaction_stream"]))
                 {
-                    managedWebSocketsApp.UseWebSockets(new WebSocketOptions() { ReplaceFeature = true });
+                    managedWebSocketsApp.UseWebSockets();
                     managedWebSocketsApp.Use(next => new TransactionStreamMiddleware(next).Invoke);
                 }
             });
diff --git a/src/Openchain.Server/project.json b/src/Openchain.Server/project.json
index ecaba36..22757a9 100644
--- a/src/Openchain.Server/project.json
+++ b/src/Openchain.Server/project.json
@@ -21,7 +21,7 @@
     "Microsoft.AspNetCore.Mvc.Core": "1.1.*",
     "Microsoft.AspNetCore.Mvc.ViewFeatures": "1.1.*",
     "Microsoft.AspNetCore.Server.Kestrel": "1.1.*",
-    "Microsoft.AspNetCore.WebSockets.Server": "0.1.0",
+    "Microsoft.AspNetCore.WebSockets": "1.0.0",
     "Microsoft.Extensions.Logging.Console": "1.1.*",
     "Microsoft.Extensions.Configuration.Json": "1.1.*",
     "Microsoft.Extensions.Configuration.UserSecrets": "1.1.*",