diff --git a/src/instance.ts b/src/instance.ts
index 0b8f522d1..e51decd69 100644
--- a/src/instance.ts
+++ b/src/instance.ts
@@ -42,7 +42,7 @@ export const createInstance = ({ app, server, options }: {
     config: customConfig
   });
 
-  wss.on("connection", (client: IClient) => {
+  wss.on("connection", (client: IClient, realm: IRealm) => {
     const messageQueue = realm.getMessageQueueById(client.getId());
 
     if (messageQueue) {
@@ -62,7 +62,7 @@ export const createInstance = ({ app, server, options }: {
     messageHandler.handle(client, message);
   });
 
-  wss.on("close", (client: IClient) => {
+  wss.on("close", (client: IClient, realm: IRealm) => {
     app.emit("disconnect", client);
   });
 
@@ -72,4 +72,4 @@ export const createInstance = ({ app, server, options }: {
 
   messagesExpire.startMessagesExpiration();
   checkBrokenConnections.start();
-};
\ No newline at end of file
+};