diff --git a/src/webpack/index.ts b/src/webpack/index.ts
index 6b057c0f6e..3e45d0af7b 100644
--- a/src/webpack/index.ts
+++ b/src/webpack/index.ts
@@ -31,11 +31,11 @@ export let isInjected = false;
 export let isReady = false;
 
 export function onInjected(listener: () => void): void {
-  internalEv.on('webpack.injected', listener);
+  internalEv.on('webpack.injected', async () => listener());
 }
 
 export function onReady(listener: () => void): void {
-  internalEv.on('webpack.ready', listener);
+  internalEv.on('webpack.ready', async () => listener());
 }
 
 export type SearchModuleCondition = (module: any, moduleId: string) => boolean;
diff --git a/src/whatsapp/misc/Socket.ts b/src/whatsapp/misc/Socket.ts
index 561c933c68..9b13e03a77 100644
--- a/src/whatsapp/misc/Socket.ts
+++ b/src/whatsapp/misc/Socket.ts
@@ -22,4 +22,4 @@ import { SocketModel } from '../models';
  */
 export declare const Socket: SocketModel;
 
-exportModule(exports, { Socket: 'Socket' }, (m) => m.Socket?.initConn);
+exportModule(exports, { Socket: 'Socket' }, (m) => m.Socket);