From dc1ba169c9fb29839b49a89ee13ba5f052894a18 Mon Sep 17 00:00:00 2001
From: Robinson <robinson@dorkbox.com>
Date: Wed, 13 Apr 2022 16:22:39 +0200
Subject: [PATCH] Fixed wrong casting of object type

---
 src/dorkbox/network/serialization/Serialization.kt | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/dorkbox/network/serialization/Serialization.kt b/src/dorkbox/network/serialization/Serialization.kt
index 4767efb1..eb45430f 100644
--- a/src/dorkbox/network/serialization/Serialization.kt
+++ b/src/dorkbox/network/serialization/Serialization.kt
@@ -27,8 +27,6 @@ import dorkbox.network.connection.streaming.StreamingControl
 import dorkbox.network.connection.streaming.StreamingControlSerializer
 import dorkbox.network.connection.streaming.StreamingData
 import dorkbox.network.connection.streaming.StreamingDataSerializer
-import dorkbox.network.connection.streaming.StreamingMessage
-import dorkbox.network.connection.streaming.StreamingState
 import dorkbox.network.handshake.HandshakeMessage
 import dorkbox.network.ping.Ping
 import dorkbox.network.ping.PingSerializer
@@ -603,11 +601,16 @@ open class Serialization<CONNECTION: Connection>(private val references: Boolean
                             }
                         }
 
-                        implClass!!
+                        // NOTE: implClass can still be null!
 
-                        logger.trace("REGISTRATION (RMI-CLIENT) ${clazz.name} -> ${implClass.name}")
+                        logger.trace {
+                            if (implClass != null) {
+                                "REGISTRATION (RMI-CLIENT) ${clazz.name} -> ${implClass.name}"
+                            } else {
+                                "REGISTRATION (RMI-CLIENT) ${clazz.name}"
+                            }
+                        }
 
-                        // implClass MIGHT BE NULL!
                         classesToRegister.add(ClassRegistrationForRmi(clazz, implClass, rmiServerSerializer))
 
                     }