diff --git a/src/framework/luafunctions.cpp b/src/framework/luafunctions.cpp
index ddaeee95..e664d8ad 100644
--- a/src/framework/luafunctions.cpp
+++ b/src/framework/luafunctions.cpp
@@ -631,6 +631,9 @@ void Application::registerLuaFunctions()
     g_lua.bindClassStaticFunction<Server>("create", &Server::create);
     g_lua.bindClassMemberFunction<Server>("acceptNext", &Server::acceptNext);
 
+    // Connection
+    g_lua.registerClass<Connection>();
+
     // Protocol
     g_lua.registerClass<Protocol>();
     g_lua.bindClassStaticFunction<Protocol>("create", []{ return ProtocolPtr(new Protocol); });
diff --git a/src/framework/net/protocol.cpp b/src/framework/net/protocol.cpp
index 90169087..6f0cc1f2 100644
--- a/src/framework/net/protocol.cpp
+++ b/src/framework/net/protocol.cpp
@@ -138,12 +138,6 @@ void Protocol::internalRecvData(uint8* buffer, uint16 size)
             g_logger.traceError("failed to decrypt message");
             return;
         }
-    } else {
-        int size = m_inputMessage->getU16();
-        if(size != m_inputMessage->getUnreadSize()) {
-            g_logger.traceError("invalid message size");
-            return;
-        }
     }
 
     onRecv(m_inputMessage);