diff --git a/src/io/netty/bootstrap/DatagramCloseListener.java b/src/dorkbox/network/connection/DatagramCloseListener.java
similarity index 81%
rename from src/io/netty/bootstrap/DatagramCloseListener.java
rename to src/dorkbox/network/connection/DatagramCloseListener.java
index 060a2de3..92e80a7f 100644
--- a/src/io/netty/bootstrap/DatagramCloseListener.java
+++ b/src/dorkbox/network/connection/DatagramCloseListener.java
@@ -13,10 +13,9 @@
  *  See the License for the specific language governing permissions and
  *  limitations under the License.
  */
-package io.netty.bootstrap;
+package dorkbox.network.connection;
 
-import dorkbox.network.connection.Connection;
-import dorkbox.network.connection.Listener;
+import io.netty.bootstrap.DatagramCloseMessage;
 
 /**
  * Uses a hint to close remote UDP connections
@@ -26,6 +25,8 @@ class DatagramCloseListener implements Listener.OnMessageReceived<Connection, Da
     @Override
     public
     void received(final Connection connection, final DatagramCloseMessage message) {
-        connection.close();
+        if (connection instanceof ConnectionImpl) {
+            ((ConnectionImpl) connection).forceClose();
+        }
     }
 }