diff --git a/src/Microsoft.AspNetCore.Sockets/Connection.cs b/src/Microsoft.AspNetCore.Sockets/Connection.cs
new file mode 100644
index 0000000000..544e1cbef3
--- /dev/null
+++ b/src/Microsoft.AspNetCore.Sockets/Connection.cs
@@ -0,0 +1,14 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using Channels;
+
+namespace Microsoft.AspNetCore.Sockets
+{
+    public class Connection
+    {
+        public string ConnectionId { get; set; }
+        public IChannel Channel { get; set; }
+    }
+}
diff --git a/src/Microsoft.AspNetCore.Sockets/ConnectionState.cs b/src/Microsoft.AspNetCore.Sockets/ConnectionState.cs
index 00e0c2929f..3386f910ec 100644
--- a/src/Microsoft.AspNetCore.Sockets/ConnectionState.cs
+++ b/src/Microsoft.AspNetCore.Sockets/ConnectionState.cs
@@ -1,5 +1,4 @@
 using System;
-using Channels;
 
 namespace Microsoft.AspNetCore.Sockets
 {
@@ -9,10 +8,4 @@ namespace Microsoft.AspNetCore.Sockets
         public bool Active { get; set; } = true;
         public Connection Connection { get; set; }
     }
-
-    public class Connection
-    {
-        public string ConnectionId { get; set; }
-        public IChannel Channel { get; set; }
-    }
 }