diff --git a/CryptoExchange.Net.UnitTests/TestImplementations/TestSocket.cs b/CryptoExchange.Net.UnitTests/TestImplementations/TestSocket.cs
index f9042ad..70f006a 100644
--- a/CryptoExchange.Net.UnitTests/TestImplementations/TestSocket.cs
+++ b/CryptoExchange.Net.UnitTests/TestImplementations/TestSocket.cs
@@ -38,6 +38,8 @@ namespace CryptoExchange.Net.UnitTests.TestImplementations
 
         public double IncomingKbps => throw new NotImplementedException();
 
+        public Uri Uri => new Uri("");
+
         public static int lastId = 0;
         public static object lastIdLock = new object();
 
@@ -111,5 +113,11 @@ namespace CryptoExchange.Net.UnitTests.TestImplementations
         {
             OnError?.Invoke(error);
         }
+
+        public async Task ProcessAsync()
+        {
+            while (Connected)
+                await Task.Delay(50);
+        }
     }
 }
diff --git a/CryptoExchange.Net/CryptoExchange.Net.csproj b/CryptoExchange.Net/CryptoExchange.Net.csproj
index fdb43bf..56a3613 100644
--- a/CryptoExchange.Net/CryptoExchange.Net.csproj
+++ b/CryptoExchange.Net/CryptoExchange.Net.csproj
@@ -6,7 +6,7 @@
     <PackageId>CryptoExchange.Net</PackageId>
     <Authors>JKorf</Authors>
     <Description>A base package for implementing cryptocurrency API's</Description>
-    <PackageVersion>5.1.7</PackageVersion>
+    <PackageVersion>5.1.7-local</PackageVersion>
     <AssemblyVersion>5.1.7</AssemblyVersion>
     <FileVersion>5.1.7</FileVersion>
     <PackageRequireLicenseAcceptance>false</PackageRequireLicenseAcceptance>
diff --git a/CryptoExchange.Net/Sockets/SocketConnection.cs b/CryptoExchange.Net/Sockets/SocketConnection.cs
index 2f24121..f9594a8 100644
--- a/CryptoExchange.Net/Sockets/SocketConnection.cs
+++ b/CryptoExchange.Net/Sockets/SocketConnection.cs
@@ -181,6 +181,15 @@ namespace CryptoExchange.Net.Sockets
             return connected;
         }
 
+        /// <summary>
+        /// Retrieve the underlying socket
+        /// </summary>
+        /// <returns></returns>
+        public IWebsocket GetSocket()
+        {
+            return _socket;
+        }
+
         /// <summary>
         /// Trigger a reconnect of the socket connection
         /// </summary>