diff --git a/src/Microsoft.AspNet.WebSockets.Server/ExtendedWebSocketAcceptContext.cs b/src/Microsoft.AspNet.WebSockets.Server/ExtendedWebSocketAcceptContext.cs
index 8b7d901aaf..6c6fa97f16 100644
--- a/src/Microsoft.AspNet.WebSockets.Server/ExtendedWebSocketAcceptContext.cs
+++ b/src/Microsoft.AspNet.WebSockets.Server/ExtendedWebSocketAcceptContext.cs
@@ -2,7 +2,7 @@
 // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
 
 using System;
-using Microsoft.AspNet.Http;
+using Microsoft.AspNet.Http.Features;
 
 namespace Microsoft.AspNet.WebSockets.Server
 {
diff --git a/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs b/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs
index 7432567bb0..8d13d9be0e 100644
--- a/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs
+++ b/src/Microsoft.AspNet.WebSockets.Server/WebSocketMiddleware.cs
@@ -8,6 +8,7 @@ using System.Net.WebSockets;
 using System.Threading.Tasks;
 using Microsoft.AspNet.Builder;
 using Microsoft.AspNet.Http;
+using Microsoft.AspNet.Http.Features;
 using Microsoft.AspNet.WebSockets.Protocol;
 
 namespace Microsoft.AspNet.WebSockets.Server