diff --git a/src/dorkbox/network/connection/EndPoint.kt b/src/dorkbox/network/connection/EndPoint.kt
index a5a2791d..07e51dc3 100644
--- a/src/dorkbox/network/connection/EndPoint.kt
+++ b/src/dorkbox/network/connection/EndPoint.kt
@@ -861,7 +861,7 @@ internal constructor(val type: Class<*>,
      *  - response manager
      *  - storage
      */
-    internal fun closeForRestart() {
+    fun closeForRestart() {
         if (shutdown.compareAndSet(expect = false, update = true)) {
             logger.info { "Shutting down for restart..." }